test(test_IVF.py): 更新模型路径和测试名称

- 更新模型路径为新的权重文件
- 修改模型名称以反映新的增强方法
This commit is contained in:
zjut 2024-11-17 15:49:42 +08:00
parent 0336fc23ba
commit 3b7b64c915

View File

@ -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-23-08.pth"
ckpt_path= r"/home/star/whaiDir/PFCFuse/models/whaiFusion11-17-10-34.pth"
for dataset_name in ["sar"]:
print("\n"*2+"="*80)
model_name="PFCFuse Enhance SCSA "
model_name="PFCFuse Enhance 增加widthblock"
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)