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

avatar: avatar_type is now only head by default

Browse Source
This commit is contained in:
Colombo
2019-12-29 19:37:59 +04:00
parent 47e539ccdd
commit efd8c16daa
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/Model_AVATAR/Model.py
+6 -6
View File
@@ -21,13 +21,13 @@ class AVATARModel(ModelBase):
#override #override
def onInitializeOptions(self, is_first_run, ask_override): def onInitializeOptions(self, is_first_run, ask_override):
if is_first_run: if is_first_run:
avatar_type = io.input_int("Avatar type ( 0:source, 1:head, 2:full_face ?:help skip:1) : ", 1, [0,1,2], #avatar_type = io.input_int("Avatar type ( 0:source, 1:head, 2:full_face ?:help skip:1) : ", 1, [0,1,2],
help_message="Training target for the model. Source is direct untouched images. Full_face or head are centered nose unaligned faces.") # help_message="Training target for the model. Source is direct untouched images. Full_face or head are centered nose unaligned faces.")
avatar_type = {0:'source', #avatar_type = {0:'source',
1:'head', # 1:'head',
2:'full_face'}[avatar_type] # 2:'full_face'}[avatar_type]
self.options['avatar_type'] = avatar_type self.options['avatar_type'] = 'head'
else: else:
self.options['avatar_type'] = self.options.get('avatar_type', 'head') self.options['avatar_type'] = self.options.get('avatar_type', 'head')