Infrared and Visible Image Fusion Based on Edge-Preserving and Attention Generative Adversarial Network

Zhu Wen-Qing,Tang Xin-Yi,Zhang Rui,Chen Xiao,Miao Zhuang
DOI: https://doi.org/10.11972/j.issn.1001-9014.2021.05.017
2021-01-01
JOURNAL OF INFRARED AND MILLIMETER WAVES
Abstract:Infrared and visible image features are quite different, and there are no ideal fused images supervise neural networks to learn the mapping relationship between the source images and the fused images, thus, the application of deep learning is limited to the field of image fusion. To solve this problem, a generative adversarial network framework based on attention mechanism and edge loss is proposed, which is applied to the infrared and visible image fusion. Derived from the thoughts of attention mechanism and adversarial training, the fusion problem is regarded as an adversarial game between the source images and the fused images, and combining channel attention and spatial attention mechanism can learn nonlinear relationship between channel domain features and spatial domain features, which enhances the expression of salient target features. At the same time, an edge-based loss function is proposed, which converts the mapping relationship between the source image pixels and the fused image pixels into the mapping relationship between the edges. Experimental results on multiple datasets demonstrate that the proposed method can effectively fuse infrared target and visible texture information, sharpen image edges, and significantly improve image clarity and contrast.
What problem does this paper attempt to address?