CalvinBackup/iperov-DeepFaceLab
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
def onInitializeOptions(self, is_first_run, ask_override):
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],
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',
1:'head',
2:'full_face'}[avatar_type]
#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.")
#avatar_type = {0:'source',
# 1:'head',
# 2:'full_face'}[avatar_type]
self.options['avatar_type'] = avatar_type
self.options['avatar_type'] = 'head'
else:
self.options['avatar_type'] = self.options.get('avatar_type', 'head')