search

millieXie / FusionMamba

FusionMamba: Dynamic Feature Enhancement for Multimodal Image Fusion with Mamba
95 stars 5 forks source link

Wrong Weight #8

Open zipper112 opened 1 month ago

zipper112 commented 1 month ago

Hello and thank you for your wonderful work. An error seems to have occurred when I loaded the weights for the infrared versus visible tasks. The naming of some modules has changed, which makes it inconsistent with the pre-trained weights.

millieXie commented 1 month ago

Please re-download the folder named "models." The error has been corrected. Thank you for bringing it to my attention!

DZSYUNNAN commented 3 weeks ago

An error seems to have occurred when I loaded the weights for the CT and MR image fusion tasks. Unexpected key(s) in state_dict: "layers.2.blocks.1.norm.weight", "layers.2.blocks.1.norm.bias"

millieXie commented 3 weeks ago

I have re-uploaded the model files. The link is included in the README file for your reference. CT-MRI: link：https://pan.baidu.com/s/1lrEiVLs2p8cMyCV-GN_ZKQ?pwd=a3v8 password：a3v8