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

saehd: removed liaech

Browse source
This commit is contained in:
Colombo 2020-03-13 13:04:26 +04:00
parent a3a493bb93
commit dde856499d
1 changed files with 2 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

5
models/Model_SAEHD/Model.py
View file

@ -61,7 +61,7 @@ class SAEHDModel(ModelBase):
resolution = np.clip ( (resolution // 16) * 16, 64, 512) resolution = np.clip ( (resolution // 16) * 16, 64, 512)
self.options['resolution'] = resolution self.options['resolution'] = resolution
self.options['face_type'] = io.input_str ("Face type", default_face_type, ['h','mf','f','wf'], help_message="Half / mid face / full face / whole face. Half face has better resolution, but covers less area of cheeks. Mid face is 30% wider than half face. 'Whole face' covers full area of face include forehead, but requires manual merge in Adobe After Effects.").lower() self.options['face_type'] = io.input_str ("Face type", default_face_type, ['h','mf','f','wf'], help_message="Half / mid face / full face / whole face. Half face has better resolution, but covers less area of cheeks. Mid face is 30% wider than half face. 'Whole face' covers full area of face include forehead, but requires manual merge in Adobe After Effects.").lower()
self.options['archi'] = io.input_str ("AE architecture", default_archi, ['df','liae','dfhd','liaehd','liaech'], help_message="'df' keeps faces more natural.\n'liae' can fix overly different face shapes.\n'hd' are experimental versions.\n'liaech' - new experimental model by @chervoniy. Based on liae, but produces more src-like face.").lower() self.options['archi'] = io.input_str ("AE architecture", default_archi, ['df','liae','dfhd','liaehd'], help_message="'df' keeps faces more natural.\n'liae' can fix overly different face shapes.\n'hd' are experimental versions.").lower()
default_d_dims = 48 if self.options['archi'] == 'dfhd' else 64 default_d_dims = 48 if self.options['archi'] == 'dfhd' else 64
default_d_dims = self.options['d_dims'] = self.load_or_def_option('d_dims', default_d_dims) default_d_dims = self.options['d_dims'] = self.load_or_def_option('d_dims', default_d_dims)
@ -169,9 +169,6 @@ class SAEHDModel(ModelBase):
self.target_dstm_all = tf.placeholder (nn.floatx, mask_shape) self.target_dstm_all = tf.placeholder (nn.floatx, mask_shape)
# Initializing model classes # Initializing model classes
if archi == 'liaech':
model_archi = nn.DeepFakeArchi(resolution, mod='chervonij')
else:
model_archi = nn.DeepFakeArchi(resolution) model_archi = nn.DeepFakeArchi(resolution)
with tf.device (models_opt_device): with tf.device (models_opt_device):