Update pre.py

ZhuLvs · Nov 27, 2024 · 7ae0985 · 7ae0985
1 parent b8d0ee1
commit 7ae0985
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)