mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-07 13:32:09 -07:00
added sumbmodules
This commit is contained in:
parent
774932a5b0
commit
5267017f7d
2 changed files with 81 additions and 49 deletions
40
core/interact/interact.txt
Normal file
40
core/interact/interact.txt
Normal 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
|
|
@ -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'])
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue