DualFuse (Pytorch)

---

An pytorch implement of "A Dual-branch Network for Infrared and Visible Image Fusion".

🧐 About

This is a pytorch implement of A Dual-branch Network for Infrared and Visible Image Fusion proposed by this paper,

https://ieeexplore.ieee.org/abstract/document/9412293

http://arxiv.org/abs/2101.09643

🎈 Usage

Quick test

run 'test.py'

Train

run 'train,py'

you need some pairs of infrared and visible light images. You can use TNO, or ,KAIST https://github.com/SoonminHwang/rgbt-ped-detection, or other.