MATCNN: Infrared and Visible Image Fusion Method Based on Multi-scale CNN with Attention Transformer
Journal:
arXiv
Published Date:
Feb 4, 2025
Abstract
While attention-based approaches have shown considerable progress in
enhancing image fusion and addressing the challenges posed by long-range
feature dependencies, their efficacy in capturing local features is compromised
by the lack of diverse receptive field extraction techniques. To overcome the
shortcomings of existing fusion methods in extracting multi-scale local
features and preserving global features, this paper proposes a novel
cross-modal image fusion approach based on a multi-scale convolutional neural
network with attention Transformer (MATCNN). MATCNN utilizes the multi-scale
fusion module (MSFM) to extract local features at different scales and employs
the global feature extraction module (GFEM) to extract global features.
Combining the two reduces the loss of detail features and improves the ability
of global feature representation. Simultaneously, an information mask is used
to label pertinent details within the images, aiming to enhance the proportion
of preserving significant information in infrared images and background
textures in visible images in fused images. Subsequently, a novel optimization
algorithm is developed, leveraging the mask to guide feature extraction through
the integration of content, structural similarity index measurement, and global
feature loss. Quantitative and qualitative evaluations are conducted across
various datasets, revealing that MATCNN effectively highlights infrared salient
targets, preserves additional details in visible images, and achieves better
fusion results for cross-modal images. The code of MATCNN will be available at
https://github.com/zhang3849/MATCNN.git.