WaveRFN-Nest: A Multi-Scale Feature Fusion Method Based on Wavelet Transform and Deep Learning

Chuan Kou,Yunxiao Zu
DOI: https://doi.org/10.1109/CSECS60003.2023.10428536
2023-01-01
Abstract:Infrared and visible images fusion is a technique that combines two types of images and has broad application prospects. With the development of deep learning, various high-performance fusion models have emerged in this field, including STDFusionNet, DRF, RFN-Nest, and others. However, most of these models focus on optimizing the model architecture and fusion performance, without considering the key features that can support subsequent tasks such as classification and object detection in the images. To address these issues, this paper proposes a multi-scale feature fusion algorithm called WaveRFN-Nest. Firstly, the spatial attention module called WaveAtten is introduced to extract important detail information from visible light images. Then, the WaveAtten module is embedded into the RFN-Nest model using a multi-scale feature fusion module, allowing the model to focus more on crucial features. The Experiments conducted on multiple datasets show that the composite model proposed in this paper ensures the quality of fused images with small parameter increments, improves the proportion of important detailed features in the images, and facilitates subsequent tasks.
What problem does this paper attempt to address?