test: 更新测试脚本以使用新的模型路径

- 修改了 pth_path 变量，使用新的模型路径 "whaiFusion11-17-16-16"
- 更新了模型名称显示，使用 pth_path 变量替代固定字符串
- 优化了测试结果输出格式，增加了换行和分隔线

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-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 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)

train.py

@@ -262,5 +262,7 @@ if True:
         'BaseFuseLayer': BaseFuseLayer.state_dict(),
         'DetailFuseLayer': DetailFuseLayer.state_dict(),
     }
-    torch.save(checkpoint, os.path.join("models/whaiFusion"+timestamp+'.pth'))
+    savepth = os.path.join("models/whaiFusion" + timestamp + '.pth');
+    torch.save(checkpoint, savepth)
+    print("save model:{}".format(savepth))