MMFF-NET

The code of "MMFF-NET: Multi-layer and multi-scale feature fusion network for low-light infrared image enhancement"

The article link for this article is below paper_link

Update

Updated code and pre-trained models：Epoch90.pth & Epoch99.pth

Usage

train.py ----Training code
test.py  ----Testing code
model.py ----You can modify the network structure

We provide two pre-trained models that you can also train on your own

Supplementary Note

This is my own code for my first post, and the overall code is not perfect.
For some reasons, our training set is not publicly available.
You can train with Zero-DCE's training set and the results are just as good.

zero-dce

Bibtex

@article{zhu2023mmff,
  title={MMFF-NET: Multi-layer and multi-scale feature fusion network for low-light infrared image enhancement},
  author={Zhu, Ge and Chen, Yuhan and Wang, Xianquan and Zhang, Yiheng},
  journal={Signal, Image and Video Processing},
  pages={1--9},
  year={2023},
  publisher={Springer}
}

Contact

If you have any questions, please contact Yuhan Chen at cyh1217552389@gmail.com