test(test_IVF.py): 更新测试模型并调整测试数据集
- 更新模型路径为 whaiFusion11-15-17-48.pth - 将测试数据集从 ["TNO","RoadScene"] 修改为 ["sar"]- 修改模型名称为 "PFCFuse 最基本版本"
This commit is contained in:
parent
344de69cb2
commit
0ef5760d76
@ -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/whaiFusion10-08-16-20.pth"
|
||||
ckpt_path= r"/home/star/whaiDir/PFCFuse/models/whaiFusion11-15-17-48.pth"
|
||||
|
||||
for dataset_name in ["TNO","RoadScene"]:
|
||||
for dataset_name in ["sar"]:
|
||||
print("\n"*2+"="*80)
|
||||
model_name="PFCFuse "
|
||||
model_name="PFCFuse 最基本版本 "
|
||||
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)
|
||||
|
Loading…
Reference in New Issue
Block a user