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}
}