|
|
|
|
@ -11,7 +11,7 @@ if cmd_opts.deepdanbooru:
|
|
|
|
|
import modules.deepbooru as deepbooru
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def preprocess(process_src, process_dst, process_width, process_height, process_flip, process_split, process_caption, process_caption_deepbooru=False):
|
|
|
|
|
def preprocess(process_src, process_dst, process_width, process_height, preprocess_txt_action, process_flip, process_split, process_caption, process_caption_deepbooru=False):
|
|
|
|
|
try:
|
|
|
|
|
if process_caption:
|
|
|
|
|
shared.interrogator.load()
|
|
|
|
|
@ -21,7 +21,7 @@ def preprocess(process_src, process_dst, process_width, process_height, process_
|
|
|
|
|
db_opts[deepbooru.OPT_INCLUDE_RANKS] = False
|
|
|
|
|
deepbooru.create_deepbooru_process(opts.interrogate_deepbooru_score_threshold, db_opts)
|
|
|
|
|
|
|
|
|
|
preprocess_work(process_src, process_dst, process_width, process_height, process_flip, process_split, process_caption, process_caption_deepbooru)
|
|
|
|
|
preprocess_work(process_src, process_dst, process_width, process_height, preprocess_txt_action, process_flip, process_split, process_caption, process_caption_deepbooru)
|
|
|
|
|
|
|
|
|
|
finally:
|
|
|
|
|
|
|
|
|
|
@ -33,7 +33,7 @@ def preprocess(process_src, process_dst, process_width, process_height, process_
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def preprocess_work(process_src, process_dst, process_width, process_height, process_flip, process_split, process_caption, process_caption_deepbooru=False):
|
|
|
|
|
def preprocess_work(process_src, process_dst, process_width, process_height, preprocess_txt_action, process_flip, process_split, process_caption, process_caption_deepbooru=False):
|
|
|
|
|
width = process_width
|
|
|
|
|
height = process_height
|
|
|
|
|
src = os.path.abspath(process_src)
|
|
|
|
|
|