search

Linfeng-Tang / SeAFusion

The code of " Image fusion in the loop of high-level vision tasks: A semantic-aware real-time infrared and visible image fusion network"
MIT License
184 stars 34 forks source link

语义分割的模型.pth文件在哪里呀 #14

Open liushh39 opened 1 year ago

Linfeng-Tang commented 1 year ago

语义分割的模型已经上传至model_final.pth

liushh39 commented 1 year ago

十分感谢

liushh39 commented 1 year ago

你好，但是运行evaluate.py出现一下错误 RuntimeError: Error(s) in loading state_dict for BiSeNet: size mismatch for conv_out.conv_out.weight: copying a param with shape torch.Size([9, 128, 1, 1]) from checkpoint, the shape in current model is torch.Size([19, 128, 1, 1]). size mismatch for conv_out16.conv_out.weight: copying a param with shape torch.Size([9, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([19, 64, 1, 1]).

Linfeng-Tang commented 1 year ago

evaluate.py和相关数据集已经更新 你可以重新下载运行。