From 890f6d5a9c5316e3b496d41dcefff93b433b4e28 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Wed, 21 Dec 2022 17:54:31 +0900 Subject: [PATCH] Fix relative path to train dir is not working. --- finetune/make_captions.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/finetune/make_captions.py b/finetune/make_captions.py index 5808051e..b02420bd 100644 --- a/finetune/make_captions.py +++ b/finetune/make_captions.py @@ -24,10 +24,13 @@ def main(args): random.seed(seed) if not os.path.exists("blip"): + args.train_data_dir = os.path.abspath(args.train_data_dir) # convert to absolute path + cwd = os.getcwd() print('Current Working Directory is: ', cwd) os.chdir('finetune') + print(f"load images from {args.train_data_dir}") 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")) print(f"found {len(image_paths)} images.")