From a435e035c000f70a42cd6c68eac890f6897dc4e9 Mon Sep 17 00:00:00 2001 From: zjut Date: Sun, 17 Nov 2024 09:53:12 +0800 Subject: [PATCH] =?UTF-8?q?test:=20=E6=9B=B4=E6=96=B0=E6=B5=8B=E8=AF=95?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E5=B9=B6=E4=BF=AE=E6=94=B9=E6=A8=A1=E5=9E=8B?= =?UTF-8?q?=E5=90=8D=E7=A7=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 更新测试模型路径为 whaiFusion11-16-23-08.pth - 修改模型名称为 "PFCFuse Enhance SCSA" --- test_IVF.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test_IVF.py b/test_IVF.py index 467f7fb..bd3c269 100644 --- a/test_IVF.py +++ b/test_IVF.py @@ -17,11 +17,11 @@ current_time = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") os.environ["CUDA_VISIBLE_DEVICES"] = "0" -ckpt_path= r"/home/star/whaiDir/PFCFuse/models/whaiFusion11-16-11-20.pth" +ckpt_path= r"/home/star/whaiDir/PFCFuse/models/whaiFusion11-16-23-08.pth" for dataset_name in ["sar"]: print("\n"*2+"="*80) - model_name="PFCFuse Enhance " + model_name="PFCFuse Enhance SCSA " print("The test result of "+dataset_name+' :') test_folder = os.path.join('test_img', dataset_name) test_out_folder=os.path.join('test_result',current_time,dataset_name)