github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-19 04:59:27 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix converterconfig

Browse source
This commit is contained in:
Colombo 2019-11-13 00:19:01 +04:00
commit 54786848bc
1 changed files with 10 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

14
converters/ConverterConfig.py
View file

@ -110,6 +110,13 @@ mode_dict = {0:'original',
7:'raw-mask-only', 7:'raw-mask-only',
8:'raw-predicted-only'} 8:'raw-predicted-only'}
mode_str_dict = {}
for key in mode_dict.keys():
mode_str_dict[ mode_dict[key] ] = key
print(mode_str_dict)
full_face_mask_mode_dict = {1:'learned', full_face_mask_mode_dict = {1:'learned',
2:'dst', 2:'dst',
3:'FAN-prd', 3:'FAN-prd',
@ -220,15 +227,14 @@ class ConverterConfigMasked(ConverterConfig):
self.export_mask_alpha = not self.export_mask_alpha self.export_mask_alpha = not self.export_mask_alpha
def ask_settings(self): def ask_settings(self):
s = """Choose mode: \n""" s = """Choose mode: \n"""
for key in mode_dict.keys(): for key in mode_dict.keys():
s += f"""({key}) {mode_dict[key]}\n""" s += f"""({key}) {mode_dict[key]}\n"""
s += f"""Default: {self.default_mode} : """ s += f"""Default: { mode_str_dict.get(self.default_mode, 1) } : """
mode = io.input_int (s, self.default_mode) mode = io.input_int (s, mode_str_dict.get(self.default_mode, 1) )
self.mode = mode_dict.get (mode, mode_dict[self.default_mode] ) self.mode = mode_dict.get (mode, self.default_mode )
if 'raw' not in self.mode: if 'raw' not in self.mode:
if self.mode == 'hist-match' or self.mode == 'hist-match-bw': if self.mode == 'hist-match' or self.mode == 'hist-match-bw':