HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Context-Aware Interaction Network for RGB-T Semantic Segmentation

Ying Lv; Zhi Liu; Gongyang Li

Context-Aware Interaction Network for RGB-T Semantic Segmentation

Abstract

RGB-T semantic segmentation is a key technique for autonomous driving scenes understanding. For the existing RGB-T semantic segmentation methods, however, the effective exploration of the complementary relationship between different modalities is not implemented in the information interaction between multiple levels. To address such an issue, the Context-Aware Interaction Network (CAINet) is proposed for RGB-T semantic segmentation, which constructs interaction space to exploit auxiliary tasks and global context for explicitly guided learning. Specifically, we propose a Context-Aware Complementary Reasoning (CACR) module aimed at establishing the complementary relationship between multimodal features with the long-term context in both spatial and channel dimensions. Further, considering the importance of global contextual and detailed information, we propose the Global Context Modeling (GCM) module and Detail Aggregation (DA) module, and we introduce specific auxiliary supervision to explicitly guide the context interaction and refine the segmentation map. Extensive experiments on two benchmark datasets of MFNet and PST900 demonstrate that the proposed CAINet achieves state-of-the-art performance. The code is available at https://github.com/YingLv1106/CAINet.

Code Repositories

yinglv1106/cainet
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
semantic-segmentation-on-nyu-depth-v2CAINet (MobileNet-V2)
Mean IoU: 52.6%
semantic-segmentation-on-syn-udtiriCAINet
IoU: 91.77
thermal-image-segmentation-on-mfn-datasetCAINet (MobileNet-V2)
mIOU: 58.6%
thermal-image-segmentation-on-pst900CAINet (MobileNet-V2)
mIoU: 84.74

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
Context-Aware Interaction Network for RGB-T Semantic Segmentation | Papers | HyperAI