HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Learning Selective Self-Mutual Attention for RGB-D Saliency Detection

{ Junwei Han Ni Zhang Nian Liu}

Learning Selective Self-Mutual Attention for RGB-D Saliency Detection

Abstract

Saliency detection on RGB-D images is receiving more and more research interests recently. Previous models adopt the early fusion or the result fusion scheme to fuse the input RGB and depth data or their saliency maps, which incur the problem of distribution gap or information loss. Some other models use the feature fusion scheme but are limited by the linear feature fusion methods. In this paper, we propose to fuse attention learned in both modalities. Inspired by the Non-local model, we integrate the self-attention and each other's attention to propagate long-range contextual dependencies, thus incorporating multi-modal information to learn attention and propagate contexts more accurately. Considering the reliability of the other modality's attention, we further propose a selection attention to weight the newly added attention term. We embed the proposed attention module in a two-stream CNN for RGB-D saliency detection. Furthermore, we also propose a residual fusion module to fuse the depth decoder features into the RGB stream. Experimental results on seven benchmark datasets demonstrate the effectiveness of the proposed model components and our final saliency model. Our code and saliency maps are available at https://github.com/nnizhang/S2MA.

Benchmarks

BenchmarkMethodologyMetrics
rgb-d-salient-object-detection-on-nju2kS2MA
Average MAE: 0.053
S-Measure: 89.4
max E-Measure: 92.7
max F-Measure: 88.9

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
Learning Selective Self-Mutual Attention for RGB-D Saliency Detection | Papers | HyperAI