Code of "SFDFusion: An Efficient Spatial-Frequency Domain Fusion Network for Infrared and Visible Image Fusion"
Accepted by ECAI 2024
paper link [here]
python 3.10
cuda 11.8
conda create -n SFD python=3.10
conda activate SFD
pip install -r requirements.txt
The training process needs wandb API key.
The config file is ./configs/cfg.yaml
python train.py
python fuse.py
MSRS dataset is used to train. You can get it from here.