github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 05:22:06 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

added sumbmodules

Browse source
This commit is contained in:
leno3003 2022-03-30 15:28:08 +02:00
parent 599aca6266
commit c40d7ee52f
3 changed files with 50 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

8
core/interact/interact.py
View file

@ -46,7 +46,7 @@ class InteractBase(object):
self.process_messages_callbacks = {} self.process_messages_callbacks = {}
self.default_answers = {} self.default_answers = {}
answer_filename = '/home/deepfake/interact_dict.pkl' answer_filename = '../../../worspace/interact/interact_dict.pkl'
if os.path.exists(answer_filename): if os.path.exists(answer_filename):
with open(answer_filename, 'rb') as file: with open(answer_filename, 'rb') as file:
self.default_answers = pickle.load(file) self.default_answers = pickle.load(file)
@ -219,9 +219,9 @@ class InteractBase(object):
def get_default_answer(self, answer_key): def get_default_answer(self, answer_key):
#Scrivere su file answer key #Scrivere su file answer key
f = open("/home/deepfake/interact.txt", "a") #f = open("/home/deepfake/interact.txt", "a")
f.write(answer_key + "\n") #f.write(answer_key + "\n")
f.close() #f.close()
if answer_key in self.default_answers: if answer_key in self.default_answers:
return self.default_answers[answer_key] return self.default_answers[answer_key]
return None return None

75
core/interact/no_interact_dict.py
View file

@ -3,44 +3,45 @@ import pickle
dictionary = { dictionary = {
'4' : '\n', '4' : '\n',
'Output image format':'png', 'Output image format':'png',
'NonInteractiveMode':'True',
'Which GPU indexes to choose?': '0', 'Which GPU indexes to choose?': '0',
'Face type': 'wf', 'Face type': 'wf',
'Max number of faces from image' : '1', 'Max number of faces from image' : '1',
'Image size' : '512', 'Image size' : '512',
'Jpeg quality' : '90', 'Jpeg quality' : '90',
'Write debug images to aligned_debug?': 'False', 'Write debug images to aligned_debug?': 'False',
'Autobackup every N hour':'2', 'Autobackup every N hour':'2',
'Write preview history' : 'False', 'Write preview history' : 'False',
'Flip SRC faces randomly':'False', 'Flip SRC faces randomly':'False',
'Flip DST faces randomly':'False', 'Flip DST faces randomly':'False',
'Batch_size': '4', 'Batch_size': '4',
'Eyes and mouth priority':'False', 'Eyes and mouth priority':'False',
'Uniform yaw distribution of samples':'True', 'Uniform yaw distribution of samples':'True',
'Blur out mask':'False', 'Blur out mask':'False',
'Place models and optimizer on GPU' : 'True', 'Place models and optimizer on GPU' : 'True',
'Use AdaBelief optimizer?' : 'True', 'Use AdaBelief optimizer?' : 'True',
'Use learning rate dropout' : 'False', 'Use learning rate dropout' : 'False',
'Enable random warp of samples' : 'True', 'Enable random warp of samples' : 'True',
'Random hue/saturation/light intensity' : '0.0', 'Random hue/saturation/light intensity' : '0.0',
'GAN power' : '0.0', 'GAN power' : '0.0',
'Face style power' : '0.0', 'Face style power' : '0.0',
'Background style power': '0.0', 'Background style power': '0.0',
'Color transfer for src faceset' : 'lct', 'Color transfer for src faceset' : 'lct',
'Enable gradient clipping': 'False', 'Enable gradient clipping': 'False',
'Enable pretraining mode' : 'False', 'Enable pretraining mode' : 'False',
'Use interactive merger?':'False', 'Use interactive merger?':'False',
'Number of workers?':'8', 'Number of workers?':'8',
'Use saved session?':'False', 'Use saved session?':'False',
'Bitrate of output file in MB/s' : '16', 'Bitrate of output file in MB/s' : '16',
'Choose erode mask modifier': '0.0', 'Choose erode mask modifier': '0.0',
'Choose blur mask modifier' : '0.0', 'Choose blur mask modifier' : '0.0',
'Choose motion blur power' : '0', 'Choose motion blur power' : '0',
'Choose output face scale modifier' : '0', 'Choose output face scale modifier' : '0',
'Choose super resolution power' : '0', 'Choose super resolution power' : '0',
'Choose image degrade by denoise power' : '0', 'Choose image degrade by denoise power' : '0',
'Choose image degrade by bicubic rescale power' : '0', 'Choose image degrade by bicubic rescale power' : '0',
'Degrade color power of final image' : '0', 'Degrade color power of final image' : '0',
'Color transfer to predicted face' : 'rct', 'Color transfer to predicted face' : 'rct',
} }
with open('/home/deepfake/interact_dict.pkl', 'wb') as handle: with open('/home/deepfake/interact_dict.pkl', 'wb') as handle:
pickle.dump(dictionary, handle, protocol=4) pickle.dump(dictionary, handle, protocol=4)

11
mainscripts/Trainer.py
View file

@ -84,11 +84,16 @@ def trainerThread (s2c, c2s, e,
if model.get_target_iter() != 0: if model.get_target_iter() != 0:
if is_reached_goal: if is_reached_goal:
io.log_info('Model already trained to target iteration. You can use preview.') 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() model_save()
os._exit(1) os._exit(1)
else: else:
io.log_info('Starting. Target iteration: %d. Press "Enter" to stop training and save model.' % ( model.get_target_iter() ) ) io.log_info('Starting. Target iteration: %d. Press "Enter" to stop training and save model.' % ( model.get_target_iter() ) )
else: else:
io.log_info('Starting. Press "Enter" to stop training and save model.') io.log_info('Starting. Press "Enter" to stop training and save model.')