From d653e594c278efb6c20d11b815778fcc50966f44 Mon Sep 17 00:00:00 2001 From: CjangCjengh <101577701+CjangCjengh@users.noreply.github.com> Date: Wed, 6 Dec 2023 09:48:42 +0800 Subject: [PATCH 1/3] fix path error --- finetune/make_captions.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/finetune/make_captions.py b/finetune/make_captions.py index b20c4106..61c766e0 100644 --- a/finetune/make_captions.py +++ b/finetune/make_captions.py @@ -13,7 +13,7 @@ import torch from torchvision import transforms from torchvision.transforms.functional import InterpolationMode sys.path.append(os.path.dirname(__file__)) -from blip.blip import blip_decoder +from blip.blip import blip_decoder, is_url import library.train_util as train_util DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu") @@ -76,6 +76,8 @@ def main(args): cwd = os.getcwd() print("Current Working Directory is: ", cwd) os.chdir("finetune") + if not is_url(args.caption_weights): + args.caption_weights = os.path.join("..", args.caption_weights) print(f"load images from {args.train_data_dir}") train_data_dir_path = Path(args.train_data_dir) From 710e777a92f3b04a0476c3e622b3a2174f3e8db9 Mon Sep 17 00:00:00 2001 From: CjangCjengh <101577701+CjangCjengh@users.noreply.github.com> Date: Fri, 8 Dec 2023 00:22:13 +0800 Subject: [PATCH 2/3] fix path error --- finetune/make_captions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetune/make_captions.py b/finetune/make_captions.py index 61c766e0..b46652c3 100644 --- a/finetune/make_captions.py +++ b/finetune/make_captions.py @@ -76,7 +76,7 @@ def main(args): cwd = os.getcwd() print("Current Working Directory is: ", cwd) os.chdir("finetune") - if not is_url(args.caption_weights): + if not is_url(args.caption_weights) and not os.path.exists(args.caption_weights): args.caption_weights = os.path.join("..", args.caption_weights) print(f"load images from {args.train_data_dir}") From d31aa143f4fb9179b18d2545d8c261c8eace88f6 Mon Sep 17 00:00:00 2001 From: CjangCjengh <101577701+CjangCjengh@users.noreply.github.com> Date: Fri, 8 Dec 2023 00:27:32 +0800 Subject: [PATCH 3/3] fix path error --- finetune/make_captions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetune/make_captions.py b/finetune/make_captions.py index b46652c3..074576bc 100644 --- a/finetune/make_captions.py +++ b/finetune/make_captions.py @@ -76,7 +76,7 @@ def main(args): cwd = os.getcwd() print("Current Working Directory is: ", cwd) os.chdir("finetune") - if not is_url(args.caption_weights) and not os.path.exists(args.caption_weights): + if not is_url(args.caption_weights) and not os.path.isfile(args.caption_weights): args.caption_weights = os.path.join("..", args.caption_weights) print(f"load images from {args.train_data_dir}")