/FusionDN

FusionDN: A Unified Densely Connected Network for Image Fusion

Primary LanguagePython

FusionDN

This is the code of the following paper (tensorflow):
@inproceedings{xu2020aaai,
title={FusionDN: A Unified Densely Connected Network for Image Fusion},
author={Xu, Han and Ma, Jiayi and Le, Zhuliang and Jiang, Junjun and Guo, Xiaojie},
booktitle={proceedings of the Thirty-Fourth AAAI Conference on Artificial Intelligence},
year={2020}
}

It is a unified model for multiple image fusion tasks, including visible and infrared image fusion, multi-exposure image fusion and multi-focus image fusion.

Framework:

Overall procedure:


Intuitive description of data flow and the process of EWC:


Fused results:

To train:

CUDA_VISIBLE_DEVICES=0,1 python main.py (2 GPUs are needed)

To test:

CUDA_VISIBLE_DEVICES=0 python test_main.py

Tips:

Large files should be downloaded separately, including the following files:

For training:

For testing: