CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactorings, added motion blur to SampleProcessor for FANSegmentator trainer

Browse Source
This commit is contained in:
iperov
2019-04-07 23:08:00 +04:00
parent a88ee7d093
commit 58d7e990f4
9 changed files with 238 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/Model_SAE/Model.py
+1 -1
View File
@@ -320,7 +320,7 @@ class SAEModel(ModelBase):
self.dst_sample_losses = []
f = SampleProcessor.TypeFlags
face_type = f.FACE_ALIGN_FULL if self.options['face_type'] == 'f' else f.FACE_ALIGN_HALF
face_type = f.FACE_TYPE_FULL if self.options['face_type'] == 'f' else f.FACE_TYPE_HALF
output_sample_types=[ [f.WARPED_TRANSFORMED | face_type | f.MODE_BGR, resolution] ]
output_sample_types += [ [f.TRANSFORMED | face_type | f.MODE_BGR, resolution // (2**i) ] for i in range(ms_count)]