added sumbmodules

This commit is contained in:
leno3003 2022-03-30 15:05:22 +02:00
parent 774932a5b0
commit 5267017f7d
2 changed files with 81 additions and 49 deletions

View file

@ -0,0 +1,40 @@
4
Output image format
4
Which GPU indexes to choose?
Face type
Max number of faces from image
Image size
Jpeg quality
Write debug images to aligned_debug?
Which GPU indexes to choose?
Face type
Image size
Jpeg quality
Which GPU indexes to choose?
Autobackup every N hour
Write preview history
Flip SRC faces randomly
Flip DST faces randomly
Batch_size
Eyes and mouth priority
Uniform yaw distribution of samples
Blur out mask
Place models and optimizer on GPU
Use AdaBelief optimizer?
Use learning rate dropout
Enable random warp of samples
Random hue/saturation/light intensity
GAN power
Face style power
Background style power
Color transfer for src faceset
Enable gradient clipping
Enable pretraining mode
Which GPU indexes to choose?
Use interactive merger?
Number of workers?
Use saved session?
Bitrate of output file in MB/s

View file

@ -1,59 +1,51 @@
import pickle import pickle
dictionary = { dictionary = {
'Output image format' : 'png', '4' : '\n',
'Override' : '0', 'Output image format':'png',
'Enable gradient clipping' : 'False', 'Which GPU indexes to choose?': '0',
' Use saved session? ' : 'False', 'Face type': 'wf',
'Enable pretraining mode' : 'False', 'Max number of faces from image' : '1',
' Press enter in 2 seconds to override model settings. ' : '\n', 'Image size' : '512',
'[0] Which GPU indexes to choose? : ' : '0', 'Jpeg quality' : '90',
'[wf] Face type ( f/wf/head ?:help ) : ' : 'wf', 'Write debug images to aligned_debug?': 'False',
'[0] Max number of faces from image ( ?:help ) : ' : '0', 'Autobackup every N hour':'2',
'[512] Image size ( 256-2048 ?:help ) : ' : '512', 'Write preview history' : 'False',
'[90] Jpeg quality ( 1-100 ?:help ) : ' : '90', 'Flip SRC faces randomly':'False',
'[n] Write debug images to aligned_debug? ( y/n ) : ' : 'False', 'Flip DST faces randomly':'False',
'[y] Continue extraction? ( y/n ?:help ) : ' : 'True', 'Batch_size': '4',
'[2] Autobackup every N hour ( 0..24 ?:help ) : ' : '2', 'Eyes and mouth priority':'False',
'[n] Write preview history ( y/n ?:help ) : ' : 'False', 'Uniform yaw distribution of samples':'True',
'[83000] Target iteration : ' : '\n', 'Blur out mask':'False',
'[n] Flip SRC faces randomly ( y/n ?:help ) : ' : 'False', 'Place models and optimizer on GPU' : 'True',
'[n] Flip DST faces randomly ( y/n ?:help ) : ' : 'False', 'Use AdaBelief optimizer?' : 'True',
'[4] Batch_size ( ?:help ) : ' : '4', 'Use learning rate dropout' : 'False',
'[n] Eyes and mouth priority ( y/n ?:help ) : ' : 'False', 'Enable random warp of samples' : 'True',
'[y] Uniform yaw distribution of samples ( y/n ?:help ) : ' : 'True', 'Random hue/saturation/light intensity' : '0.0',
'[n] Blur out mask ( y/n ?:help ) : ' : 'False', 'GAN power' : '0.0',
'[y] Place models and optimizer on GPU ( y/n ?:help ) : ' : 'True', 'Face style power' : '0.0',
'[y] Use AdaBelief optimizer? ( y/n ?:help ) : ' : 'True', 'Background style power': '0.0',
'[n] Use learning rate dropout ( n/y/cpu ?:help ) : ' : 'n', 'Color transfer for src faceset' : 'lct',
'[y] Enable random warp of samples ( y/n ?:help ) : ' : 'True', 'Enable gradient clipping': 'False',
'[0.0] Random hue/saturation/light intensity ( 0.0 .. 0.3 ?:help ) : ' : '0.0', 'Enable pretraining mode' : 'False',
'[0.0] GAN power ( 0.0 .. 5.0 ?:help ) : ' : '0.0', 'Use interactive merger?':'False',
'[0.0] Face style power ( 0.0..100.0 ?:help ) : ' : '0.0', 'Number of workers?':'8',
'[0.0] Background style power ( 0.0..100.0 ?:help ) : ' : '0.0', 'Use saved session?':'False',
'[lct] Color transfer for src faceset ( none/rct/lct/mkl/idt/sot ?:help ) : ' : 'lct', 'Bitrate of output file in MB/s' : '16',
'[n] Use interactive merger? ( y/n ) : ' : 'n', 'Choose erode mask modifier',
'[1] 2 : ' : '1', 'Choose blur mask modifier',
'[1] 3 : ' : '4', 'Choose motion blur power' : '0',
'[0] Choose erode mask modifier ( -400..400 ) : ' : '100', 'Choose output face scale modifier' : '0',
'[0] Choose blur mask modifier ( 0..400 ) : ' : '150', 'Choose super resolution power' : '0',
'[0] Choose motion blur power ( 0..100 ) : ' : '0', 'Choose image degrade by denoise power' : '0',
'[0] Choose output face scale modifier ( -50..50 ) : ' : '0', 'Choose image degrade by bicubic rescale power' : '0',
'[0] 1 ( ?:help ) : ' : '0', 'Degrade color power of final image' : '0',
'[0] Choose super resolution power ( 0..100 ?:help ) : ' : '0', 'Color transfer to predicted face' : 'rct',
'[0] Choose image degrade by denoise power ( 0..500 ) : ' : '0',
'[0] Choose image degrade by bicubic rescale power ( 0..100 ) : ' : '0',
'[0] Degrade color power of final image ( 0..100 ) : ' : '0',
'Color transfer to predicted face ( rct/lct/mkl/mkl-m/idt/idt-m/sot-m/mix-m ) : ' : 'rct',
'[8] Number of workers? ( 1-8 ?:help ) : ' : '8',
'[16] Bitrate of output file in MB/s : ' : '16',
} }
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)
with open('/home/deepfake/interact_dict.pkl', 'rb') as handle: with open('/home/deepfake/interact_dict.pkl', 'rb') as handle:
d = pickle.load(handle) d = pickle.load(handle)
s = "Use saved"
res = dict(filter(lambda item: s in item[0], d.items()))
print(list(res.values())[0]) print(d['Color transfer to predicted face'])