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

Fix relative path to train dir is not working.

Browse Source
This commit is contained in:
Kohya S
2022-12-21 17:54:31 +09:00
parent dfbecbc4d7
commit 890f6d5a9c
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
finetune/make_captions.py
View File

@@ -24,10 +24,13 @@ def main(args):
random.seed(seed) random.seed(seed)
if not os.path.exists("blip"): if not os.path.exists("blip"):
args.train_data_dir = os.path.abspath(args.train_data_dir) # convert to absolute path
cwd = os.getcwd() cwd = os.getcwd()
print('Current Working Directory is: ', cwd) print('Current Working Directory is: ', cwd)
os.chdir('finetune') os.chdir('finetune')
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, "*.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.")