mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-07 05:22:06 -07:00
added sumbmodules
This commit is contained in:
parent
05850a227c
commit
dc39e72fb9
2 changed files with 20 additions and 24 deletions
|
@ -219,23 +219,16 @@ class InteractBase(object):
|
|||
self.key_events[wnd_name] = []
|
||||
return ar
|
||||
|
||||
def input(self, s, def_ans=" "):
|
||||
|
||||
#f = open("/home/deepfake/interact.txt", "a")
|
||||
#f.write(s + def_ans + "\n")
|
||||
#f.close()
|
||||
print("La stringa è: |", s.strip(), "|")
|
||||
def input(self, s, answer_key):
|
||||
if "WARNING " in s:
|
||||
return "\n"
|
||||
|
||||
res = dict(filter(lambda item: s in item[0], self.default_answers.items()))
|
||||
key = list(res.keys())[0]
|
||||
if key in self.default_answers:
|
||||
return list(res.values())[0]
|
||||
if answer_key in self.default_answers:
|
||||
return self.default_answers[answer_key]
|
||||
return input(s)
|
||||
|
||||
def input_number(self, s, default_value, valid_list=None, show_default_value=True, add_info=None, help_message=None):
|
||||
s_base = s
|
||||
def input_number(self, s, default_value, valid_list=None, show_default_value=True, add_info=None, help_message=None, answer_key=None):
|
||||
answer_key = answer_key if answer_key is not None else s
|
||||
if show_default_value and default_value is not None:
|
||||
s = f"[{default_value}] {s}"
|
||||
|
||||
|
@ -256,7 +249,7 @@ class InteractBase(object):
|
|||
|
||||
while True:
|
||||
try:
|
||||
inp = self.input(s_base, str(default_value))
|
||||
inp = self.input(s, answer_key)
|
||||
if len(inp) == 0:
|
||||
result = default_value
|
||||
break
|
||||
|
@ -278,7 +271,8 @@ class InteractBase(object):
|
|||
print(result)
|
||||
return result
|
||||
|
||||
def input_int(self, s, default_value, valid_range=None, valid_list=None, add_info=None, show_default_value=True, help_message=None):
|
||||
def input_int(self, s, default_value, valid_range=None, valid_list=None, add_info=None, show_default_value=True, help_message=None, answer_key=None):
|
||||
answer_key = answer_key if answer_key is not None else s
|
||||
if show_default_value:
|
||||
if len(s) != 0:
|
||||
s_base = s
|
||||
|
@ -309,7 +303,7 @@ class InteractBase(object):
|
|||
|
||||
while True:
|
||||
try:
|
||||
inp = self.input(s_base, str(default_value))
|
||||
inp = self.input(s, answer_key)
|
||||
if len(inp) == 0:
|
||||
raise ValueError("")
|
||||
|
||||
|
@ -332,7 +326,8 @@ class InteractBase(object):
|
|||
print (result)
|
||||
return result
|
||||
|
||||
def input_bool(self, s, default_value, help_message=None):
|
||||
def input_bool(self, s, default_value, help_message=None, answer_key=None):
|
||||
answer_key = answer_key if answer_key is not None else s
|
||||
s_base = s
|
||||
s = f"[{yn_str[default_value]}] {s} ( y/n"
|
||||
|
||||
|
@ -342,7 +337,7 @@ class InteractBase(object):
|
|||
|
||||
while True:
|
||||
try:
|
||||
inp = self.input(s_base, str(default_value))
|
||||
inp = self.input(s_base, answer_key)
|
||||
if len(inp) == 0:
|
||||
raise ValueError("")
|
||||
|
||||
|
@ -355,7 +350,8 @@ class InteractBase(object):
|
|||
print ( "y" if default_value else "n" )
|
||||
return default_value
|
||||
|
||||
def input_str(self, s, default_value, valid_list=None, show_default_value=True, help_message=None):
|
||||
def input_str(self, s, default_value, valid_list=None, show_default_value=True, help_message=None, answer_key=None):
|
||||
answer_key = answer_key if answer_key is not None else s
|
||||
s_base = s
|
||||
if show_default_value and default_value is not None:
|
||||
s = f"[{default_value}] {s}"
|
||||
|
@ -377,10 +373,9 @@ class InteractBase(object):
|
|||
s += " : "
|
||||
|
||||
|
||||
print("Stringa costruita: |", s, "|")
|
||||
while True:
|
||||
try:
|
||||
inp = self.input(s_base, default_value)
|
||||
inp = self.input(s_base, answer_key)
|
||||
|
||||
if len(inp) == 0:
|
||||
if default_value is None:
|
||||
|
@ -419,7 +414,8 @@ class InteractBase(object):
|
|||
except:
|
||||
sq.put (False)
|
||||
|
||||
def input_in_time (self, str, max_time_sec):
|
||||
def input_in_time (self, str, max_time_sec, answer_key=None):
|
||||
answer_key = answer_key if answer_key is not None else s
|
||||
sq = multiprocessing.Queue()
|
||||
p = multiprocessing.Process(target=self.input_process, args=( sys.stdin.fileno(), sq, str))
|
||||
p.daemon = True
|
||||
|
|
|
@ -36,7 +36,7 @@ class MergerConfig(object):
|
|||
for key in self.sharpen_dict.keys():
|
||||
s += f"""({key}) {self.sharpen_dict[key]}\n"""
|
||||
io.log_info(s)
|
||||
self.sharpen_mode = io.input_int ("1", 0, valid_list=self.sharpen_dict.keys(), help_message="Enhance details by applying sharpen filter.")
|
||||
self.sharpen_mode = io.input_int ("", 0, valid_list=self.sharpen_dict.keys(), help_message="Enhance details by applying sharpen filter.", "1")
|
||||
|
||||
if self.sharpen_mode != 0:
|
||||
self.blursharpen_amount = np.clip ( io.input_int ("Choose blur/sharpen amount", 0, add_info="-100..100"), -100, 100 )
|
||||
|
@ -193,7 +193,7 @@ class MergerConfigMasked(MergerConfig):
|
|||
for key in mode_dict.keys():
|
||||
s += f"""({key}) {mode_dict[key]}\n"""
|
||||
io.log_info(s)
|
||||
mode = io.input_int ("2", mode_str_dict.get(self.default_mode, 1) )
|
||||
mode = io.input_int ("", mode_str_dict.get(self.default_mode, 1) , "2")
|
||||
|
||||
self.mode = mode_dict.get (mode, self.default_mode )
|
||||
|
||||
|
@ -208,7 +208,7 @@ class MergerConfigMasked(MergerConfig):
|
|||
for key in mask_mode_dict.keys():
|
||||
s += f"""({key}) {mask_mode_dict[key]}\n"""
|
||||
io.log_info(s)
|
||||
self.mask_mode = io.input_int ("3", 1, valid_list=mask_mode_dict.keys() )
|
||||
self.mask_mode = io.input_int ("", 1, valid_list=mask_mode_dict.keys(), "3" )
|
||||
|
||||
if 'raw' not in self.mode:
|
||||
self.erode_mask_modifier = np.clip ( io.input_int ("Choose erode mask modifier", 0, add_info="-400..400"), -400, 400)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue