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.
Overall procedure:
Intuitive description of data flow and the process of EWC:
CUDA_VISIBLE_DEVICES=0,1 python main.py (2 GPUs are needed)
CUDA_VISIBLE_DEVICES=0 python test_main.py
Large files should be downloaded separately, including the following files:
- Training dataset
- vgg16 (put it in 'VGGnet/')