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

add .bmp support for wd14 tagger

Browse Source
This commit is contained in:
breakcore2
2023-01-06 04:07:04 -08:00
parent fac1813ac0
commit 4735b21318
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
finetune/tag_images_by_wd14_tagger.py
View File

@@ -37,7 +37,9 @@ def main(args):
# 画像を読み込む
image_paths = glob.glob(os.path.join(args.train_data_dir, "*.jpg")) + \
glob.glob(os.path.join(args.train_data_dir, "*.png")) + glob.glob(os.path.join(args.train_data_dir, "*.webp"))
glob.glob(os.path.join(args.train_data_dir, "*.png")) + \
glob.glob(os.path.join(args.train_data_dir, "*.webp")) + \
glob.glob(os.path.join(args.train_data_dir, "*.bmp"))
print(f"found {len(image_paths)} images.")
print("loading model and labels")