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

fix: #53

Browse Source
This commit is contained in:
unknown
2023-01-29 22:30:45 +09:00
parent 67e698af67
commit c20745b6e8
5 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
finetune/make_captions.py
View File

@@ -31,7 +31,7 @@ def main(args):
os.chdir('finetune') os.chdir('finetune')
print(f"load images from {args.train_data_dir}") print(f"load images from {args.train_data_dir}")
image_paths = glob.glob(os.path.join(args.train_data_dir, "*.jpg")) + \ image_paths = glob.glob(os.path.join(args.train_data_dir, "*.jpg")) + glob.glob(os.path.join(args.train_data_dir, "*.jpeg")) + \
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"))
print(f"found {len(image_paths)} images.") print(f"found {len(image_paths)} images.")

5
finetune/merge_captions_to_metadata.py
View File

@@ -10,7 +10,7 @@ from tqdm import tqdm
def main(args): def main(args):
image_paths = glob.glob(os.path.join(args.train_data_dir, "*.jpg")) + \ image_paths = glob.glob(os.path.join(args.train_data_dir, "*.jpg")) + glob.glob(os.path.join(args.train_data_dir, "*.jpeg")) + \
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"))
print(f"found {len(image_paths)} images.") print(f"found {len(image_paths)} images.")
@@ -30,7 +30,8 @@ def main(args):
for image_path in tqdm(image_paths): for image_path in tqdm(image_paths):
caption_path = os.path.splitext(image_path)[0] + args.caption_extension caption_path = os.path.splitext(image_path)[0] + args.caption_extension
with open(caption_path, "rt", encoding='utf-8') as f: with open(caption_path, "rt", encoding='utf-8') as f:
caption = f.readlines()[0].strip() lines = f.readlines()
caption = lines[0].strip() if len(lines) > 0 else ""
image_key = image_path if args.full_path else os.path.splitext(os.path.basename(image_path))[0] image_key = image_path if args.full_path else os.path.splitext(os.path.basename(image_path))[0]
if image_key not in metadata: if image_key not in metadata:

2
finetune/merge_dd_tags_to_metadata.py
View File

@@ -10,7 +10,7 @@ from tqdm import tqdm
def main(args): def main(args):
image_paths = glob.glob(os.path.join(args.train_data_dir, "*.jpg")) + \ image_paths = glob.glob(os.path.join(args.train_data_dir, "*.jpg")) + glob.glob(os.path.join(args.train_data_dir, "*.jpeg")) + \
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"))
print(f"found {len(image_paths)} images.") print(f"found {len(image_paths)} images.")

2
finetune/prepare_buckets_latents.py
View File

@@ -36,7 +36,7 @@ def get_latents(vae, images, weight_dtype):
def main(args): def main(args):
image_paths = glob.glob(os.path.join(args.train_data_dir, "*.jpg")) + \ image_paths = glob.glob(os.path.join(args.train_data_dir, "*.jpg")) + glob.glob(os.path.join(args.train_data_dir, "*.jpeg")) + \
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"))
print(f"found {len(image_paths)} images.") print(f"found {len(image_paths)} images.")

2
finetune/tag_images_by_wd14_tagger.py
View File

@@ -36,7 +36,7 @@ def main(args):
args.model_dir, SUB_DIR), force_download=True, force_filename=file) args.model_dir, SUB_DIR), force_download=True, force_filename=file)
# 画像を読み込む # 画像を読み込む
image_paths = glob.glob(os.path.join(args.train_data_dir, "*.jpg")) + \ image_paths = glob.glob(os.path.join(args.train_data_dir, "*.jpg")) + glob.glob(os.path.join(args.train_data_dir, "*.jpeg")) + \
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"))
print(f"found {len(image_paths)} images.") print(f"found {len(image_paths)} images.")