1580158
/
stable-diffusion-webui-dataset-tag-editor
mirror of https://github.com/toshiaki1729/stable-diffusion-webui-dataset-tag-editor.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

first aid for #60

pull/64/head
toshiaki1729 2023-03-22 23:24:22 +09:00
parent cb8f3a72f4
commit 518406286d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/dataset_tag_editor/dte_logic.py
View File

@ -18,7 +18,7 @@ WD_TAGGER_THRESHOLDS = [0.35, 0.35, 0.3537, 0.3685, 0.3771] # v1: idk if it's ok
INTERROGATORS = [captioning.BLIP(), tagger.DeepDanbooru()] + [tagger.WaifuDiffusion(name, WD_TAGGER_THRESHOLDS[i]) for i, name in enumerate(WD_TAGGER_NAMES)]
INTERROGATOR_NAMES = [it.name() for it in INTERROGATORS]
re_tags = re.compile(r'^(.+?)( \[\d+\])?$')
re_tags = re.compile(r'^([\s\S]+?)( \[\d+\])?$')
def interrogate_image(path:str, interrogator_name:str, threshold_booru, threshold_wd):