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

- 修改了 pth_path 变量,使用新的模型路径 "whaiFusion11-17-16-16"
- 更新了模型名称显示,使用 pth_path 变量替代固定字符串
- 优化了测试结果输出格式,增加了换行和分隔线
This commit is contained in:
zjut 2024-11-18 09:42:51 +08:00
parent 8383c68ff7
commit 73b398b811

View File

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