Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

SwinFuse: A Residual Swin Transformer Fusion Network for Infrared and Visible Images

About

The existing deep learning fusion methods mainly concentrate on the convolutional neural networks, and few attempts are made with transformer. Meanwhile, the convolutional operation is a content-independent interaction between the image and convolution kernel, which may lose some important contexts and further limit fusion performance. Towards this end, we present a simple and strong fusion baseline for infrared and visible images, namely\textit{ Residual Swin Transformer Fusion Network}, termed as SwinFuse. Our SwinFuse includes three parts: the global feature extraction, fusion layer and feature reconstruction. In particular, we build a fully attentional feature encoding backbone to model the long-range dependency, which is a pure transformer network and has a stronger representation ability compared with the convolutional neural networks. Moreover, we design a novel feature fusion strategy based on $L_{1}$-norm for sequence matrices, and measure the corresponding activity levels from row and column vector dimensions, which can well retain competitive infrared brightness and distinct visible details. Finally, we testify our SwinFuse with nine state-of-the-art traditional and deep learning methods on three different datasets through subjective observations and objective comparisons, and the experimental results manifest that the proposed SwinFuse obtains surprising fusion performance with strong generalization ability and competitive computational efficiency. The code will be available at https://github.com/Zhishe-Wang/SwinFuse.

Zhishe Wang, Yanlin Chen, Wenyu Shao, Hui Li, Lei Zhang• 2022

Related benchmarks

TaskDatasetResultRank
Visible-Infrared Image FusionMSRS (test)
Average Gradient (AG)1.93
43
Infrared-Visible Image FusionLLVIP (test)
EN5.84
23
Medical image fusionMRI-PET (test)
Entropy (EN)2.91
16
Medical image fusionMRI-SPECT
Entropy (EN)2.79
13
Medical image fusionMRI-CT (test)
EN3.94
13
Image FusionMSRS (test)
Inference Time (s)0.116
10
Image FusionM3FD
PC0.3936
10
Image FusionTNO (test)
Inference Time (s)0.128
10
Image FusionRoadScene (test)
Inference Time (s)0.071
10
Image FusionRoadScene
PC0.3858
10
Showing 10 of 11 rows

Other info

Follow for update