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 committed 2019-04-07 23:08:00 +04:00
1 parent a88ee7d093
commit 58d7e990f4
9 files changed
+238 -72
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

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)]