whaifree/pfcfuse
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(PFCFuse): 更新模型权重并修改测试名称

Browse Source 
- 更新模型权重文件路径为 whaiFusion11-16-21-39.pth
- 修改模型测试名称为 "PFCFuse Enhance 不同FusionLayer"
This commit is contained in:
zjut 2024-11-17 09:55:03 +08:00
parent aae81d97fd
commit 7ad3d631ff
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test_IVF.py
View File

@ -17,11 +17,11 @@ current_time = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
os.environ["CUDA_VISIBLE_DEVICES"] = "0" os.environ["CUDA_VISIBLE_DEVICES"] = "0"
ckpt_path= r"/home/star/whaiDir/PFCFuse/models/whaiFusion11-16-11-20.pth" ckpt_path= r"/home/star/whaiDir/PFCFuse/models/whaiFusion11-16-21-39.pth"
for dataset_name in ["sar"]: for dataset_name in ["sar"]:
print("\n"*2+"="*80) print("\n"*2+"="*80)
model_name="PFCFuse Enhance " model_name="PFCFuse Enhance 不同FusionLayer"
print("The test result of "+dataset_name+' :') print("The test result of "+dataset_name+' :')
test_folder = os.path.join('test_img', 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',current_time,dataset_name)