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
210 stars 37 forks source link

你好，但是运行evaluate.py出现以下错误 #15

Open liushh39 opened 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，你可以尝试重新下载这个project 并运行看看