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

default avatar type option is now HEAD. Head produces stable result.

Browse source
This commit is contained in:
iperov 2019-08-25 08:16:54 +04:00
parent d129b5dd7f
commit 6c66495bcf
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
models/Model_AVATAR/Model.py
View file

@ -23,15 +23,15 @@ 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:full_face, 2:head ?:help skip:0) : ", 0, [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:'full_face', 1:'head',
2:'head'}[avatar_type] 2:'full_face'}[avatar_type]
self.options['avatar_type'] = avatar_type self.options['avatar_type'] = avatar_type
else: else:
self.options['avatar_type'] = self.options.get('avatar_type', 'source') self.options['avatar_type'] = self.options.get('avatar_type', 'head')
if is_first_run or ask_override: if is_first_run or ask_override:
def_stage = self.options.get('stage', 1) def_stage = self.options.get('stage', 1)