added sumbmodules

This commit is contained in:
leno3003 2022-03-31 11:16:41 +02:00
parent 726fc41e60
commit 764a44a467

View file

@ -295,23 +295,21 @@ class ModelBase(object):
self.choose_preview_history = io.input_bool("Randomly choose new image for preview history", False, help_message="Preview image history will stay stuck with old faces if you reuse the same model on different celebs. Choose no unless you are changing src/dst to a new person") self.choose_preview_history = io.input_bool("Randomly choose new image for preview history", False, help_message="Preview image history will stay stuck with old faces if you reuse the same model on different celebs. Choose no unless you are changing src/dst to a new person")
def ask_target_iter(self, default_value=0): def ask_target_iter(self, default_value=0):
print("Path in ask_target_iter") answer_filename = 'workspace/interact/interact_dict.pkl'
cmd = "pwd" if os.path.exists(answer_filename):
os.system(cmd) with open(answer_filename, 'rb') as file:
#answer_filename = 'workspace/interact/interact_dict.pkl' parameters = pickle.load(file)
#if os.path.exists(answer_filename): if parameters['NoInteractiveMode']:
# with open(answer_filename, 'rb') as file: print("Entro in ask_target_iter")
# parameters = pickle.load(file) default_target_iter = self.load_or_def_option('target_iter', default_value + 10000)
# if parameters['NoInteractiveMode']: self.options['target_iter'] = max(0, default_target_iter + 10000)
# default_target_iter = self.load_or_def_option('target_iter', default_value + 10000) print("TARGET ITERATION: " + str(self.options['target_iter']))
# self.options['target_iter'] = max(0, default_target_iter + 10000) else:
# print("TARGET ITERATION: " + str(self.options['target_iter'])) default_target_iter = self.load_or_def_option('target_iter', default_value)
# else: self.options['target_iter'] = max(0, io.input_int("Target iteration", default_target_iter))
# default_target_iter = self.load_or_def_option('target_iter', default_value) else:
# self.options['target_iter'] = max(0, io.input_int("Target iteration", default_target_iter)) default_target_iter = self.load_or_def_option('target_iter', default_value)
#else: self.options['target_iter'] = max(0, io.input_int("Target iteration", default_target_iter))
default_target_iter = self.load_or_def_option('target_iter', default_value)
self.options['target_iter'] = max(0, io.input_int("Target iteration", default_target_iter))
def ask_random_flip(self): def ask_random_flip(self):
default_random_flip = self.load_or_def_option('random_flip', True) default_random_flip = self.load_or_def_option('random_flip', True)