From e09e1f166dd7267e551622b622cef567caa43b0d Mon Sep 17 00:00:00 2001 From: leno3003 Date: Mon, 28 Mar 2022 16:18:15 +0200 Subject: [PATCH] automatized arguments with pickle --- core/interact/interact_dict.py | 51 ++++++++++++++++++++++++++++++++++ 1 file changed, 51 insertions(+) create mode 100644 core/interact/interact_dict.py diff --git a/core/interact/interact_dict.py b/core/interact/interact_dict.py new file mode 100644 index 0000000..7c487e7 --- /dev/null +++ b/core/interact/interact_dict.py @@ -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)