S4Fusion: Saliency-aware Selective State Space Model for Infrared Visible Image Fusion

Environment

Python 3.11

Pytorch 2.1.0

mamba-ssm 1.2.0

Test

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.py

the fused images will be saved in ./results

Train

The train code and whole training dataset will be public available after Sep 29 for academic and research purposes only.

Results