From f28bb255f4e5d269a8e127d3cd8c642f12d47c61 Mon Sep 17 00:00:00 2001 From: zjut Date: Mon, 18 Nov 2024 09:47:16 +0800 Subject: [PATCH] =?UTF-8?q?build(.idea):=20=E6=9B=B4=E6=96=B0=E9=A1=B9?= =?UTF-8?q?=E7=9B=AE=E9=85=8D=E7=BD=AE=E5=B9=B6=E8=B0=83=E6=95=B4=E6=B5=8B?= =?UTF-8?q?=E8=AF=95=E8=84=9A=E6=9C=AC?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 更新 ProjectRootManager 配置,切换到本地 Python 3.8环境 - 修改模块配置,使用继承的 JDK - 更新测试脚本,使用变量 pth_path 代替硬编码的模型路径- 优化测试结果输出目录结构 --- .idea/PFCFuse.iml | 2 +- .idea/misc.xml | 2 +- test_IVF.py | 8 +++++--- 3 files changed, 7 insertions(+), 5 deletions(-) diff --git a/.idea/PFCFuse.iml b/.idea/PFCFuse.iml index a8d74d0..8b8c395 100644 --- a/.idea/PFCFuse.iml +++ b/.idea/PFCFuse.iml @@ -2,7 +2,7 @@ - + diff --git a/.idea/misc.xml b/.idea/misc.xml index d3caf2e..1586a88 100644 --- a/.idea/misc.xml +++ b/.idea/misc.xml @@ -12,5 +12,5 @@ - + \ No newline at end of file diff --git a/test_IVF.py b/test_IVF.py index ef78c63..472c4cd 100644 --- a/test_IVF.py +++ b/test_IVF.py @@ -17,14 +17,16 @@ 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-17-10-34.pth" +pth_path = "whaiFusion11-17-20-18" +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 Enhance 增加widthblock" + 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',current_time,dataset_name) + test_out_folder=os.path.join('test_result',pth_path,dataset_name) device = 'cuda' if torch.cuda.is_available() else 'cpu' Encoder = nn.DataParallel(Restormer_Encoder()).to(device)