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

#7

pull/8/head
toshiaki1729 2022-11-13 11:46:24 +09:00
parent 4a14c054ea
commit ba5c22c1b0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/dataset_tag_editor/dataset_tag_editor.py
View File

@ -151,7 +151,7 @@ class DatasetTagEditor:
return (img_paths, tag_set)
def replace_tags(self, search_tags: List[str], replace_tags: List[str], filter_tags: Optional[List[str]] = None, append_to_begin: bool = False):
def replace_tags(self, search_tags: List[str], replace_tags: List[str], filter_tags: Optional[List[str]] = None, prepend: bool = False):
img_paths, _ = self.get_filtered_imgpath_and_tags(filter_tags=filter_tags)
tags_to_append = replace_tags[len(search_tags):]
tags_to_remove = search_tags[len(replace_tags):]
@ -164,7 +164,7 @@ class DatasetTagEditor:
for img_path in img_paths:
tags_removed = [t for t in self.img_tag_dict.get(img_path) if t not in tags_to_remove]
tags_replaced = [tags_to_replace.get(t) if t in tags_to_replace.keys() else t for t in tags_removed]
self.set_tags_by_image_path(img_path, tags_to_append + tags_replaced if append_to_begin else tags_replaced + tags_to_append)
self.set_tags_by_image_path(img_path, tags_to_append + tags_replaced if prepend else tags_replaced + tags_to_append)
self.construct_tag_counts()