david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SAE: changed base erode/blur mask size for converter

Browse Source
This commit is contained in:
iperov
2019-01-07 17:46:50 +04:00
parent 104651b292
commit 75f7a9575a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/Model_SAE/Model.py
+2 -2
View File
@@ -231,8 +231,8 @@ class SAEModel(ModelBase):
def get_converter(self, **in_options): def get_converter(self, **in_options):
from models import ConverterMasked from models import ConverterMasked
base_erode_mask_modifier = 30 if self.options['face_type'] == 'f' else 100 base_erode_mask_modifier = 40 if self.options['face_type'] == 'f' else 100
base_blur_mask_modifier = 0 if self.options['face_type'] == 'f' else 100 base_blur_mask_modifier = 10 if self.options['face_type'] == 'f' else 100
face_type = FaceType.FULL if self.options['face_type'] == 'f' else FaceType.HALF face_type = FaceType.FULL if self.options['face_type'] == 'f' else FaceType.HALF