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.

Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
STDFusionNet_Results		STDFusionNet_Results
Test_ir		Test_ir
Test_vi		Test_vi
Train_ir		Train_ir
Train_ir_mask		Train_ir_mask
Train_ir_mask_blur		Train_ir_mask_blur
Train_vi		Train_vi
checkpoint		checkpoint
LICENSE		LICENSE
README.md		README.md
model.py		model.py
test_network.py		test_network.py
test_one_image.py		test_one_image.py
train.py		train.py
train_network.py		train_network.py
utils.py		utils.py

Provide feedback