STDFusionNet

The code of "STDFusionNet: An Infrared and Visible Image Fusion Network Based on Salient Target Detection"

To Train:

Run "CUDA_VISIBLE_DEVICES=0 python train.py" to train the network.

To Test:

Run "CUDA_VISIBLE_DEVICES=0 python test_one_image.py" to test the network.

If this work is helpful to you, please cite it as：

@article{ma2021STDFusionNet,
  title={STDFusionNet: An Infrared and Visible Image Fusion Network Based on Salient Target Detection},
  author={Jiayi Ma, Linfeng Tang, Meilong Xu, Hao Zhang, and Guobao Xiao},
  journal={IEEE Transactions on Instrumentation and Measurement},
  year={2021},
  volume={70},
  number={},
  pages={1-13},
  doi={10.1109/TIM.2021.3075747}，
  publisher={IEEE}
}