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

fix

Browse source
This commit is contained in:
Colombo 2019-09-19 11:20:30 +04:00
parent c06d073936
commit a325d0353b
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
models/Model_TrueFace/Model.py
View file

@ -38,7 +38,7 @@ class TrueFaceModel(ModelBase):
self.set_vram_batch_requirements({2:1,3:1,4:4,5:8,6:16}) self.set_vram_batch_requirements({2:1,3:1,4:4,5:8,6:16})
resolution = self.options['resolution'] resolution = self.options['resolution']
face_type = FaceType.FULL if self.options['face_type'] == 'f' else FaceType.HALF face_type = self.face_type = FaceType.FULL if self.options['face_type'] == 'f' else FaceType.HALF
self.model = FUNIT( face_type_str=FaceType.toString(face_type), self.model = FUNIT( face_type_str=FaceType.toString(face_type),
batch_size=self.batch_size, batch_size=self.batch_size,
@ -169,12 +169,11 @@ class TrueFaceModel(ModelBase):
#override #override
def get_ConverterConfig(self): def get_ConverterConfig(self):
face_type = FaceType.FULL
import converters import converters
return self.predictor_func, (self.options['resolution'], self.options['resolution'], 3), converters.ConverterConfigMasked(face_type=face_type, return self.predictor_func, (self.options['resolution'], self.options['resolution'], 3), converters.ConverterConfigMasked(face_type=self.face_type,
default_mode = 1, default_mode = 1,
clip_hborder_mask_per=0.0625 if (face_type == FaceType.FULL) else 0, clip_hborder_mask_per=0.0625 if (self.face_type == FaceType.FULL) else 0,
) )
Model = TrueFaceModel Model = TrueFaceModel