학술논문

AcFusion: Infrared and Visible Image Fusion Based on Self-Attention and Convolution With Enhanced Information Extraction
Document Type
Periodical
Source
IEEE Transactions on Consumer Electronics IEEE Trans. Consumer Electron. Consumer Electronics, IEEE Transactions on. 70(1):4155-4167 Feb, 2024
Subject
Power, Energy and Industry Applications
Components, Circuits, Devices and Systems
Fields, Waves and Electromagnetics
Feature extraction
Transformers
Image fusion
Convolution
Task analysis
Object detection
Generators
Multi-head self-attention
CNN
image fusion
gradient residual dense
Language
ISSN
0098-3063
1558-4127
Abstract
The purpose of fusing infrared and visible images is to create a single fused image that not only includes texture details and prominent objects but also being well-suited for further intelligent processing. Existing fusion methods often use local convolution, which fails to account for multi-scale and global feature dependencies, while Transformer-based approaches typically face constraints related to computational resources and input image size. To this end, we propose an innovative and streamlined fusion network, called AcFusion. Firstly, we introduce ACmix, which consists of a combination of convolution and multi-head self-attention, to enhance the global modeling capability while retaining as much meaningful information as possible from the source image for further processing. We design the Sobel operator-based attention gradient block (SWGD) to suppress loss of high-frequency information and enhance fine-grained information of the mode. Furthermore, we design a Residual-Dense Attention Block (RDAB) using SWGD as a component, which further improves the representation ability of features through residual connection. We conduct generalization experiments and ablation experiments to demonstrate the effectiveness of our fusion method in integrating information from different modalities. Moreover, we also verify the superiority of our method in the downstream task of object detection. The code is available at: https://github.com/Ist-Zhy/AcFusion.