github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 21:12:07 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

2
models/Model_SAE/Model.py
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)]