github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-08 05:51:40 -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 17:03:36 +02:00
parent 05ebcfdef2
commit 21024c64ad
2 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
core/interact/interact.py
View file

@ -224,7 +224,7 @@ class InteractBase(object):
#f = open("/home/deepfake/interact.txt", "a") #f = open("/home/deepfake/interact.txt", "a")
#f.write(s + def_ans + "\n") #f.write(s + def_ans + "\n")
#f.close() #f.close()
print("La stringa è: ", s) print("La stringa è: |", s.strip(), "|")
if "WARNING " in s: if "WARNING " in s:
return "\n" return "\n"
@ -371,6 +371,7 @@ class InteractBase(object):
s += " : " s += " : "
print("Stringa costruita: |", s, "|")
while True: while True:
try: try:
inp = self.input(s, default_value) inp = self.input(s, default_value)

5
core/interact/interact_dict.py
View file

@ -34,7 +34,7 @@ dictionary = {
'[n] Enable pretraining mode ( y/n ?:help ) : ' : 'False', '[n] Enable pretraining mode ( y/n ?:help ) : ' : 'False',
'[n] Use interactive merger? ( y/n ) : ' : 'False', '[n] Use interactive merger? ( y/n ) : ' : 'False',
'[1] 2 : ' : '1', '[1] 2 : ' : '1',
'[1] 3 : ' : '1', '[1] 3 : ' : '4',
'[0] Choose erode mask modifier ( -400..400 ) : ' : '150', '[0] Choose erode mask modifier ( -400..400 ) : ' : '150',
'[0] Choose blur mask modifier ( 0..400 ) : ' : '150', '[0] Choose blur mask modifier ( 0..400 ) : ' : '150',
'[0] Choose motion blur power ( 0..100 ) : ' : '0', '[0] Choose motion blur power ( 0..100 ) : ' : '0',
@ -44,8 +44,9 @@ dictionary = {
'[0] Choose image degrade by denoise power ( 0..500 ) : ' : '0', '[0] Choose image degrade by denoise power ( 0..500 ) : ' : '0',
'[0] Choose image degrade by bicubic rescale power ( 0..100 ) : ' : '0', '[0] Choose image degrade by bicubic rescale power ( 0..100 ) : ' : '0',
'[0] Degrade color power of final image ( 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', '[8] Number of workers? ( 1-8 ?:help ) : ' : '8',
'[16] Bitrate of output file in MB/s : ' : '16', '[16] Bitrate of output file in MB/s : ' : '16',
} }
with open('interact_dict.pkl', 'wb') as handle: with open('/home/deepfake/interact_dict.pkl', 'wb') as handle:
pickle.dump(dictionary, handle, protocol=pickle.HIGHEST_PROTOCOL) pickle.dump(dictionary, handle, protocol=pickle.HIGHEST_PROTOCOL)