STDFusionNet

March 25, 2021 · View on GitHub

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.