added sumbmodules

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

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)

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.')