Attention based dual UNET network for infrared and visible image fusion

Xuejiao Wang,Zhen Hua,Jinjiang Li
DOI: https://doi.org/10.1007/s11042-024-18196-x
IF: 2.577
2024-01-23
Multimedia Tools and Applications
Abstract:How to accurately extract the effective information and profile features of source images has been a difficult problem in the domain of infrared and visible image fusion. The existing fusion algorithms lack scientific decision-making in feature information extraction and fusion weight allocation, which can easily result in the loss of information or redundancy and affect the fusion effect. Therefore, this paper proposes a dual U-Net network based on the attention mechanism to realize the fusion of infrared and visible images. The method integrates the feature information of two different structural U-Net branches. The inner nested connected network extracts and fuses multi-scale infrared and visible features, while the attention mechanism is introduced in the decoding block to enhance the model's attention to the significant features of the image to generate rich semantic information. The outer U-Net network extracts the depth features of the fused image, enhances the contour features of the fused image, and improves the retention of spatial feature. Comparison with six existing fusion methods on a public dataset shows that the algorithm in this paper is superior in both subjective vision and objective evaluation.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering
What problem does this paper attempt to address?