HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Explicit Attention-Enhanced Fusion for RGB-Thermal Perception Tasks

Mingjian Liang; Junjie Hu; Chenyu Bao; Hua Feng; Fuqin Deng; Tin Lun Lam

Explicit Attention-Enhanced Fusion for RGB-Thermal Perception Tasks

Abstract

Recently, RGB-Thermal based perception has shown significant advances. Thermal information provides useful clues when visual cameras suffer from poor lighting conditions, such as low light and fog. However, how to effectively fuse RGB images and thermal data remains an open challenge. Previous works involve naive fusion strategies such as merging them at the input, concatenating multi-modality features inside models, or applying attention to each data modality. These fusion strategies are straightforward yet insufficient. In this paper, we propose a novel fusion method named Explicit Attention-Enhanced Fusion (EAEF) that fully takes advantage of each type of data. Specifically, we consider the following cases: i) both RGB data and thermal data, ii) only one of the types of data, and iii) none of them generate discriminative features. EAEF uses one branch to enhance feature extraction for i) and iii) and the other branch to remedy insufficient representations for ii). The outputs of two branches are fused to form complementary features. As a result, the proposed fusion method outperforms state-of-the-art by 1.6\% in mIoU on semantic segmentation, 3.1\% in MAE on salient object detection, 2.3\% in mAP on object detection, and 8.1\% in MAE on crowd counting. The code is available at https://github.com/FreeformRobotics/EAEFNet.

Code Repositories

Benchmarks

BenchmarkMethodologyMetrics
thermal-image-segmentation-on-mfn-datasetEAEFNet (ResNet-152)
mIOU: 58.9
thermal-image-segmentation-on-mfn-datasetEAFFNet (ResNet-50)
mIOU: 55.9
thermal-image-segmentation-on-noisy-rs-rgb-tEAEFNet
mIoU: 60.0

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Explicit Attention-Enhanced Fusion for RGB-Thermal Perception Tasks | Papers | HyperAI