HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Efficient RGB-D Semantic Segmentation for Indoor Scene Analysis

Daniel Seichter Mona Köhler Benjamin Lewandowski Tim Wengefeld Horst-Michael Gross

Efficient RGB-D Semantic Segmentation for Indoor Scene Analysis

Abstract

Analyzing scenes thoroughly is crucial for mobile robots acting in different environments. Semantic segmentation can enhance various subsequent tasks, such as (semantically assisted) person perception, (semantic) free space detection, (semantic) mapping, and (semantic) navigation. In this paper, we propose an efficient and robust RGB-D segmentation approach that can be optimized to a high degree using NVIDIA TensorRT and, thus, is well suited as a common initial processing step in a complex system for scene analysis on mobile robots. We show that RGB-D segmentation is superior to processing RGB images solely and that it can still be performed in real time if the network architecture is carefully designed. We evaluate our proposed Efficient Scene Analysis Network (ESANet) on the common indoor datasets NYUv2 and SUNRGB-D and show that we reach state-of-the-art performance while enabling faster inference. Furthermore, our evaluation on the outdoor dataset Cityscapes shows that our approach is suitable for other areas of application as well. Finally, instead of presenting benchmark results only, we also show qualitative results in one of our indoor application scenarios.

Code Repositories

evilpanda009/rain-perception
pytorch
Mentioned in GitHub
TUI-NICR/ESANet
Official
pytorch
Mentioned in GitHub
Barchid/RGBD-Seg
pytorch
Mentioned in GitHub
tui-nicr/nicr-scene-analysis-datasets
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
semantic-segmentation-on-cityscapesESANet-R34-NBt1D
Mean IoU (class): 80.09%
semantic-segmentation-on-nyu-depth-v2ESANet (R18-NBt1D )
Mean IoU: 48.17
semantic-segmentation-on-nyu-depth-v2ESANet (R34-NBt1D)
Mean IoU: 50.30
semantic-segmentation-on-sun-rgbdCMX (B5)
Mean IoU: 48.17
semantic-segmentation-on-thud-robotic-datasetESANet
mIoU: 78.42
semantic-segmentation-on-urbanlfESANet
mIoU (Real): n.a.
mIoU (Syn): 79.43
thermal-image-segmentation-on-rgb-t-glassESANet
MAE: 0.040

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
Efficient RGB-D Semantic Segmentation for Indoor Scene Analysis | Papers | HyperAI