VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

show warning if no caption and no class token

Browse Source
This commit is contained in:
Kohya S
2023-05-30 22:53:50 +09:00
parent 226db64736
commit 990ceddd14
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
library/train_util.py
View File

@@ -1087,8 +1087,9 @@ class DreamBoothDataset(BaseDataset):
for img_path in img_paths: for img_path in img_paths:
cap_for_img = read_caption(img_path, subset.caption_extension) cap_for_img = read_caption(img_path, subset.caption_extension)
if cap_for_img is None and subset.class_tokens is None: if cap_for_img is None and subset.class_tokens is None:
print(f"neither caption file nor class tokens are found. use empty caption for {img_path}") print(f"neither caption file nor class tokens are found. use empty caption for {img_path} / キャプションファイルもclass tokenも見つかりませんでした。空のキャプションを使用します: {img_path}")
captions.append("") captions.append("")
missing_captions.append(img_path)
else: else:
if cap_for_img is None: if cap_for_img is None:
captions.append(subset.class_tokens) captions.append(subset.class_tokens)
@@ -1103,10 +1104,12 @@ class DreamBoothDataset(BaseDataset):
number_of_missing_captions_to_show = 5 number_of_missing_captions_to_show = 5
remaining_missing_captions = number_of_missing_captions - number_of_missing_captions_to_show remaining_missing_captions = number_of_missing_captions - number_of_missing_captions_to_show
print(f"No caption file found for {number_of_missing_captions} images. Training will continue without captions for these images") print(
f"No caption file found for {number_of_missing_captions} images. Training will continue without captions for these images. If class token exists, it will be used. / {number_of_missing_captions}枚の画像にキャプションファイルが見つかりませんでした。これらの画像についてはキャプションなしで学習を続行します。class tokenが存在する場合はそれを使います。"
)
for i, missing_caption in enumerate(missing_captions): for i, missing_caption in enumerate(missing_captions):
if i >= number_of_missing_captions_to_show: if i >= number_of_missing_captions_to_show:
print(missing_caption+f"... and {remaining_missing_captions} more") print(missing_caption + f"... and {remaining_missing_captions} more")
break break
print(missing_caption) print(missing_caption)
return img_paths, captions return img_paths, captions