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

Update Model.py

Browse source 
in the line 30 Avatar_type section takes directly 'head' as avatar type, so that I commented it and uncommented lines 24 to 28
This commit is contained in:
zakaria47fs 2020-01-24 13:15:38 +01:00 • committed by GitHub
commit af20983d31
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
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'] = 'head' #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')
@ -487,4 +487,4 @@ class AVATARModel(ModelBase):
return func return func
Model = AVATARModel Model = AVATARModel