From 5b8c56ef7fa6fd61a02df696a51bc68087f57ca5 Mon Sep 17 00:00:00 2001 From: ZhuLvs Date: Thu, 28 Nov 2024 01:32:49 +0800 Subject: [PATCH] Update pre.py --- model/pre.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model/pre.py b/model/pre.py index f1cdeb6..96dcf74 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" - output_folder = "../input/out" + img_folder = "./input/testA" + output_folder = "./input/out" predict_and_save(model, img_folder, val_transform, device, output_folder)