LENFusion: A Joint Low-Light Enhancement and Fusion Network for Nighttime Infrared and Visible Image Fusion
This is official code of "LENFusion: A Joint Low-Light Enhancement and Fusion Network for Nighttime Infrared and Visible Image Fusion".
conda env create -f LENFusion.yaml
During training, eval is used for evaluation, and model is used to store the training weight.
The code has two training datasets in the train folder:
- LFN_traingdata is the training dataset of Luminance Feedback Network (LFN). Links: google drive and baidu.
- vi and ir are the training dataset of the main networks (Luminance Adjustment Network and Re-enhancement Fusion Network, LAN and RFN). Links: google drive and baidu.
Download checkpoint at the links: google drive and baidu. The test data from the KAIST, LLVIP, MSRS, TNO datasets.
numpy=1.26.0
opencv-python=4.8.1.78
python=3.10.13
torch=1.12.1
torchvision=0.13.1
@article{Chen2024LENFusion,
title={LENFusion: A Joint Low-Light Enhancement and Fusion Network for Nighttime Infrared and Visible Image Fusion},
author={Chen, Jun and Yang, Liling and Liu, Wei and Tian, Xin and Ma, Jiayi},
journal={IEEE Transactions on Instrumentation and Measurement},
volume={},
pages={},
year={2024}
}