Home

Awesome

FusionMamba

Paper

Get Started

Dataset

Installation

  1. Clone the repository:
git clone https://github.com/PSRben/FusionMamba.git
  1. Install the Mamba implementation by following the instructions in the Mamba-block directory.

  2. Install other packages:

pip install einops h5py opencv-python torchinfo scipy numpy

Usage

# train
python train.py --train_data_path ./path_to_data/train_WV3.h5 --val_data_path ./path_to_data/valid_WV3.h5
# test
python test.py --file_path ./path_to_data/name.h5 --save_dir ./path_to_dir --weight ./weights/epochs.pth

Citation

@misc{peng2024fusionmamba,
      title={FusionMamba: Efficient Image Fusion with State Space Model}, 
      author={Siran Peng and Xiangyu Zhu and Haoyu Deng and Zhen Lei and Liang-Jian Deng},
      year={2024},
      eprint={2404.07932},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

Contact

We are glad to hear from you. If you have any questions, please feel free to contact siran_peng@163.com.