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

fix #45

pull/48/head v0.1.0
toshiaki1729 2023-02-28 19:27:40 +09:00
parent 7191240d59
commit 8f1f045e59
2 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/dataset_tag_editor/dataset_tag_editor.py
View File

@ -464,16 +464,18 @@ class DatasetTagEditor:
except:
continue
else:
abs_path = str(img_path.absolute())
if not use_temp_dir:
img.already_saved_as = str(img_path.absolute())
images[img_path] = img
img.already_saved_as = abs_path
images[abs_path] = img
imgpaths.append(img_path)
imgpaths.append(abs_path)
return imgpaths, images
def load_captions(imgpaths: List[Path]):
def load_captions(imgpaths: List[str]):
taglists = []
for img_path in imgpaths:
for abs_path in imgpaths:
img_path = Path(abs_path)
text_path = img_path.with_suffix(caption_ext)
caption_text = ''
if interrogate_method != InterrogateMethod.OVERWRITE:

7
scripts/dataset_tag_editor/kohya-ss_finetune_metadata.py
View File

@ -31,7 +31,7 @@ def write(dataset, dataset_dir, out_path, in_path=None, overwrite=False, save_as
for data in dataset.datas.values():
img_path, tags = Path(data.imgpath), data.tags
img_key = str(img_path) if use_full_path else img_path.stem
img_key = str(img_path.absolute()) if use_full_path else img_path.stem
save_caption = ', '.join(tags) if save_as_caption else tags
if img_key not in result:
@ -65,6 +65,8 @@ def read(dataset_dir, json_path, use_temp_dir:bool):
for image_key, img_md in metadata.items():
img_path = Path(image_key)
abs_path = None
img = None
if img_path.is_file():
abs_path, img = load_image(img_path)
if abs_path is None or img is None:
@ -77,7 +79,8 @@ def read(dataset_dir, json_path, use_temp_dir:bool):
continue
images[abs_path] = img
break
if abs_path is None or img is None:
continue
caption = img_md.get('caption')
tags = img_md.get('tags')
if tags is None: