diff --git a/core/interact/interact.py b/core/interact/interact.py index e20e4c9..7ad0d1c 100644 --- a/core/interact/interact.py +++ b/core/interact/interact.py @@ -46,7 +46,7 @@ class InteractBase(object): self.process_messages_callbacks = {} self.default_answers = {} - answer_filename = '/home/deepfake/interact_dict.pkl' + answer_filename = '../../../worspace/interact/interact_dict.pkl' if os.path.exists(answer_filename): with open(answer_filename, 'rb') as file: self.default_answers = pickle.load(file) @@ -219,9 +219,9 @@ class InteractBase(object): def get_default_answer(self, answer_key): #Scrivere su file answer key - f = open("/home/deepfake/interact.txt", "a") - f.write(answer_key + "\n") - f.close() + #f = open("/home/deepfake/interact.txt", "a") + #f.write(answer_key + "\n") + #f.close() if answer_key in self.default_answers: return self.default_answers[answer_key] return None diff --git a/core/interact/no_interact_dict.py b/core/interact/no_interact_dict.py index e3b8e10..2922625 100644 --- a/core/interact/no_interact_dict.py +++ b/core/interact/no_interact_dict.py @@ -3,44 +3,45 @@ import pickle dictionary = { '4' : '\n', 'Output image format':'png', + 'NonInteractiveMode':'True', 'Which GPU indexes to choose?': '0', -'Face type': 'wf', -'Max number of faces from image' : '1', -'Image size' : '512', -'Jpeg quality' : '90', -'Write debug images to aligned_debug?': 'False', -'Autobackup every N hour':'2', -'Write preview history' : 'False', -'Flip SRC faces randomly':'False', -'Flip DST faces randomly':'False', -'Batch_size': '4', -'Eyes and mouth priority':'False', -'Uniform yaw distribution of samples':'True', -'Blur out mask':'False', -'Place models and optimizer on GPU' : 'True', -'Use AdaBelief optimizer?' : 'True', -'Use learning rate dropout' : 'False', -'Enable random warp of samples' : 'True', -'Random hue/saturation/light intensity' : '0.0', -'GAN power' : '0.0', -'Face style power' : '0.0', -'Background style power': '0.0', -'Color transfer for src faceset' : 'lct', -'Enable gradient clipping': 'False', -'Enable pretraining mode' : 'False', -'Use interactive merger?':'False', -'Number of workers?':'8', -'Use saved session?':'False', -'Bitrate of output file in MB/s' : '16', -'Choose erode mask modifier': '0.0', -'Choose blur mask modifier' : '0.0', -'Choose motion blur power' : '0', -'Choose output face scale modifier' : '0', -'Choose super resolution power' : '0', -'Choose image degrade by denoise power' : '0', -'Choose image degrade by bicubic rescale power' : '0', -'Degrade color power of final image' : '0', -'Color transfer to predicted face' : 'rct', + 'Face type': 'wf', + 'Max number of faces from image' : '1', + 'Image size' : '512', + 'Jpeg quality' : '90', + 'Write debug images to aligned_debug?': 'False', + 'Autobackup every N hour':'2', + 'Write preview history' : 'False', + 'Flip SRC faces randomly':'False', + 'Flip DST faces randomly':'False', + 'Batch_size': '4', + 'Eyes and mouth priority':'False', + 'Uniform yaw distribution of samples':'True', + 'Blur out mask':'False', + 'Place models and optimizer on GPU' : 'True', + 'Use AdaBelief optimizer?' : 'True', + 'Use learning rate dropout' : 'False', + 'Enable random warp of samples' : 'True', + 'Random hue/saturation/light intensity' : '0.0', + 'GAN power' : '0.0', + 'Face style power' : '0.0', + 'Background style power': '0.0', + 'Color transfer for src faceset' : 'lct', + 'Enable gradient clipping': 'False', + 'Enable pretraining mode' : 'False', + 'Use interactive merger?':'False', + 'Number of workers?':'8', + 'Use saved session?':'False', + 'Bitrate of output file in MB/s' : '16', + 'Choose erode mask modifier': '0.0', + 'Choose blur mask modifier' : '0.0', + 'Choose motion blur power' : '0', + 'Choose output face scale modifier' : '0', + 'Choose super resolution power' : '0', + 'Choose image degrade by denoise power' : '0', + 'Choose image degrade by bicubic rescale power' : '0', + 'Degrade color power of final image' : '0', + 'Color transfer to predicted face' : 'rct', } with open('/home/deepfake/interact_dict.pkl', 'wb') as handle: pickle.dump(dictionary, handle, protocol=4) diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index 97412bd..b9fe779 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -84,11 +84,16 @@ def trainerThread (s2c, c2s, e, if model.get_target_iter() != 0: if is_reached_goal: io.log_info('Model already trained to target iteration. You can use preview.') - if os.path.exists("/home/deepfake/interact_dict.pkl"): + + answer_filename = '../../../worspace/interact/interact_dict.pkl' + if os.path.exists(answer_filename): + with open(answer_filename, 'rb') as file: + parameters = pickle.load(file) + if parameters['NoInteractiveMode']: model_save() os._exit(1) - else: - io.log_info('Starting. Target iteration: %d. Press "Enter" to stop training and save model.' % ( model.get_target_iter() ) ) + else: + io.log_info('Starting. Target iteration: %d. Press "Enter" to stop training and save model.' % ( model.get_target_iter() ) ) else: io.log_info('Starting. Press "Enter" to stop training and save model.')