From a325d0353bf078e5ad49a21e86f2068a6ad38a72 Mon Sep 17 00:00:00 2001 From: Colombo Date: Thu, 19 Sep 2019 11:20:30 +0400 Subject: [PATCH] fix --- models/Model_TrueFace/Model.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/models/Model_TrueFace/Model.py b/models/Model_TrueFace/Model.py index b4c605b..883be09 100644 --- a/models/Model_TrueFace/Model.py +++ b/models/Model_TrueFace/Model.py @@ -38,7 +38,7 @@ class TrueFaceModel(ModelBase): self.set_vram_batch_requirements({2:1,3:1,4:4,5:8,6:16}) 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), batch_size=self.batch_size, @@ -169,12 +169,11 @@ class TrueFaceModel(ModelBase): #override def get_ConverterConfig(self): - face_type = FaceType.FULL 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, - 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