Python 3.11
Pytorch 2.1.0
mamba-ssm 1.2.0
change the params of test.py and perform:
trained_model: Our trained model
ir_path: Infrared images path
vi_path: Visible images path
save_path: fused images path
python test.pythe fused images will be saved in ./results
The train code and whole training dataset will be public available after Sep 29 for academic and research purposes only.
