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

automatized arguments with pickle

Browse source
This commit is contained in:
leno3003 2022-03-28 16:18:15 +02:00
parent a5fad7d8d5
commit e09e1f166d
1 changed files with 51 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

51
core/interact/interact_dict.py Normal file
View file

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