From 7ae09859ea1ff28daeff835117794b4b3e19d7e8 Mon Sep 17 00:00:00 2001 From: ZhuLvs Date: Thu, 28 Nov 2024 01:38:58 +0800 Subject: [PATCH] Update pre.py --- model/pre.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/pre.py b/model/pre.py index d18017c..fc93833 100644 --- a/model/pre.py +++ b/model/pre.py @@ -47,7 +47,7 @@ def remove_module_prefix(state_dict): transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]), ]) - img_folder = "./input/testA" + img_folder = "./input/valA" output_folder = "./input/out" predict_and_save(model, img_folder, val_transform, device, output_folder)