This commit is contained in:
iperov 2019-01-10 23:01:33 +04:00
parent 127cc3303a
commit e39d1a2619
2 changed files with 4 additions and 4 deletions

View file

@ -36,13 +36,13 @@ class SAEModel(ModelBase):
self.options['lighter_encoder'] = self.options.get('lighter_encoder', False)
if is_first_run or ask_override:
default_style_power = 100 if is_first_run else self.options['face_style_power']
default_style_power = 100 if is_first_run else self.options.get('face_style_power', 100)
self.options['face_style_power'] = np.clip ( input_int("Face style power (0..100 ?:help skip:%d) : " % (default_style_power), default_style_power, help_message="How fast NN will learn dst face style during generalization of src and dst faces."), 0, 100 )
else:
self.options['face_style_power'] = self.options.get('face_style_power', 100)
if is_first_run or ask_override:
default_style_power = 100 if is_first_run else self.options['bg_style_power']
default_style_power = 100 if is_first_run else self.options.get('bg_style_power', 100)
self.options['bg_style_power'] = np.clip ( input_int("Background style power (0..100 ?:help skip:%d) : " % (default_style_power), default_style_power, help_message="How fast NN will learn dst background style during generalization of src and dst faces."), 0, 100 )
else:
self.options['bg_style_power'] = self.options.get('bg_style_power', 100)