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

change code using booru since it's modules are deprecated 

#9
pull/41/head
toshiaki1729 2022-11-22 19:23:56 +09:00
parent 42f8e8a30b
commit 3868def72f
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/dataset_tag_editor/dataset_tag_editor.py
View File

@ -233,7 +233,7 @@ class DatasetTagEditor:
interrogate_text += shared.interrogator.generate_caption(img)
if use_booru:
tmp = deepbooru.get_tags_from_process(img)
tmp = deepbooru.model.tag_multi(img)
interrogate_text += (', ' if interrogate_text and tmp else '') + tmp
if interrogate_method == InterrogateMethod.OVERWRITE:
@ -251,9 +251,7 @@ class DatasetTagEditor:
if use_clip:
shared.interrogator.load()
if use_booru:
db_opts = deepbooru.create_deepbooru_opts()
db_opts[deepbooru.OPT_INCLUDE_RANKS] = False
deepbooru.create_deepbooru_process(opts.interrogate_deepbooru_score_threshold, db_opts)
deepbooru.model.start()
load_images(filepath_set = filepath_set)
@ -262,7 +260,7 @@ class DatasetTagEditor:
if use_clip:
shared.interrogator.send_blip_to_ram()
if use_booru:
deepbooru.release_process()
deepbooru.model.stop()
self.construct_tag_counts()
self.set_img_filter_img_path()