github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 05:22:06 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

12
models/Model_AVATAR/Model.py
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')