HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

SPSN: Superpixel Prototype Sampling Network for RGB-D Salient Object Detection

Minhyeok Lee; Chaewon Park; Suhwan Cho; Sangyoun Lee

SPSN: Superpixel Prototype Sampling Network for RGB-D Salient Object Detection

Abstract

RGB-D salient object detection (SOD) has been in the spotlight recently because it is an important preprocessing operation for various vision tasks. However, despite advances in deep learning-based methods, RGB-D SOD is still challenging due to the large domain gap between an RGB image and the depth map and low-quality depth maps. To solve this problem, we propose a novel superpixel prototype sampling network (SPSN) architecture. The proposed model splits the input RGB image and depth map into component superpixels to generate component prototypes. We design a prototype sampling network so that the network only samples prototypes corresponding to salient objects. In addition, we propose a reliance selection module to recognize the quality of each RGB and depth feature map and adaptively weight them in proportion to their reliability. The proposed method makes the model robust to inconsistencies between RGB images and depth maps and eliminates the influence of non-salient objects. Our method is evaluated on five popular datasets, achieving state-of-the-art performance. We prove the effectiveness of the proposed method through comparative experiments.

Code Repositories

Hydragon516/SPSN
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
rgb-d-salient-object-detection-on-desSPSN
Average MAE: 0.016
S-Measure: 93.8
max E-Measure: 97.6
max F-Measure: 94.3
rgb-d-salient-object-detection-on-nju2kSPSN
Average MAE: 0.032
S-Measure: 91.8
max E-Measure: 95.0
max F-Measure: 92.0
rgb-d-salient-object-detection-on-nlprSPSN
Average MAE: 0.022
S-Measure: 92.6
max E-Measure: 96.2
max F-Measure: 91.4
rgb-d-salient-object-detection-on-sipSPSN
Average MAE: 0.042
S-Measure: 89.2
max E-Measure: 93.4
max F-Measure: 89.9
rgb-d-salient-object-detection-on-stereSPSN
Average MAE: 0.035
S-Measure: 90.7
max E-Measure: 94.3
max F-Measure: 90.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