SFDFusion

Code of "SFDFusion: An Efficient Spatial-Frequency Domain Fusion Network For Infrared and Visible Image Fusion"

Environments

python 3.10
cuda 11.8

Install

conda create -n SFD python=3.10
conda activate SFD
pip install -r requirements.txt

Train

The training process needs wandb API key. The config file is ./configs/cfg.yaml

python train.py

Inference

python fuse.py

Dataset

MSRS dataset is used to train. You can get it from here.