From 73b398b811a63b8118f688dc5190c8be3c4cf054 Mon Sep 17 00:00:00 2001 From: zjut Date: Mon, 18 Nov 2024 09:42:51 +0800 Subject: [PATCH] =?UTF-8?q?test:=20=E6=9B=B4=E6=96=B0=E6=B5=8B=E8=AF=95?= =?UTF-8?q?=E8=84=9A=E6=9C=AC=E4=BB=A5=E4=BD=BF=E7=94=A8=E6=96=B0=E7=9A=84?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E8=B7=AF=E5=BE=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 修改了 pth_path 变量，使用新的模型路径 "whaiFusion11-17-16-16" - 更新了模型名称显示，使用 pth_path 变量替代固定字符串 - 优化了测试结果输出格式，增加了换行和分隔线 --- test_IVF.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test_IVF.py b/test_IVF.py index 09ad867..8990564 100644 --- a/test_IVF.py +++ b/test_IVF.py @@ -17,13 +17,13 @@ current_time = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") os.environ["CUDA_VISIBLE_DEVICES"] = "0" -pth_path = "whaiFusion11-15-17-48" +pth_path = "whaiFusion11-17-16-16" ckpt_path= r"/home/star/whaiDir/PFCFuse/models/"+pth_path+".pth" print("path_pth:{}".format(ckpt_path)) for dataset_name in ["sar"]: print("\n"*2+"="*80) - model_name="PFCFuse 最基本版本 " + model_name=pth_path print("The test result of "+dataset_name+' :') test_folder = os.path.join('test_img', dataset_name) test_out_folder=os.path.join('test_result',pth_path,dataset_name)