Semantic Segmentation and Scene Reconstruction of RGB-D Image Frames: An End-to-End Modular Pipeline for Robotic Applications

📅 2024-10-23
📈 Citations: 0
✨ Influential: 0
📄 PDF
🤖 AI Summary
To address insufficient geometric-semantic joint understanding of robots in unstructured environments, this paper proposes an end-to-end modular RGB-D understanding pipeline. Our method introduces a novel hybrid mask generation mechanism integrating SAM2 with a lightweight semantic classifier for pixel-level semantic segmentation and instance awareness; incorporates ReID-enhanced cross-frame human tracking and semantic-weighted TSDF point cloud fusion to ensure geometric fidelity and semantic consistency; and outputs structured scene representations in USD format. Evaluated on ADE20K, our approach achieves 47.0% mIoU—surpassing SegFormer and OneFormer in boundary accuracy—and attains a reconstruction error of only 25.3 mm. It runs 1.81× faster than prior methods and has been validated for deployment feasibility on real-world Kinect data.

Technology Category

Intelligent Robots: Multimodal Perception & Sensor FusionComputer Vision: SegmentationKnowledge Representation and Reasoning: Geometric, Spatial, and Temporal Reasoning

Application Category

Semantics and Knowledge: Methods to enhance, augment, integrate or synergize semantic models such as knowledge graphs and LLMsSearch and Retrieval-Augmented AI: Retrieval-Augmented Generation (RAG) and multi-modal RAGEconomics, Online Markets and Human Computation: Trust and reliance of crowd workers and data experts on GenAI
📝 Abstract
Robots operating in unstructured environments require a comprehensive understanding of their surroundings, necessitating geometric and semantic information from sensor data. Traditional RGB-D processing pipelines focus primarily on geometric reconstruction, limiting their ability to support advanced robotic perception, planning, and interaction. A key challenge is the lack of generalized methods for segmenting RGB-D data into semantically meaningful components while maintaining accurate geometric representations. We introduce a novel end-to-end modular pipeline that integrates state-of-the-art semantic segmentation, human tracking, point-cloud fusion, and scene reconstruction. Our approach improves semantic segmentation accuracy by leveraging the foundational segmentation model SAM2 with a hybrid method that combines its mask generation with a semantic classification model, resulting in sharper masks and high classification accuracy. Compared to SegFormer and OneFormer, our method achieves a similar semantic segmentation accuracy (mIoU of 47.0% vs 45.9% in the ADE20K dataset) but provides much more precise object boundaries. Additionally, our human tracking algorithm interacts with the segmentation enabling continuous tracking even when objects leave and re-enter the frame by object re-identification. Our point cloud fusion approach reduces computation time by 1.81x while maintaining a small mean reconstruction error of 25.3 mm by leveraging the semantic information. We validate our approach on benchmark datasets and real-world Kinect RGB-D data, demonstrating improved efficiency, accuracy, and usability. Our structured representation, stored in the Universal Scene Description (USD) format, supports efficient querying, visualization, and robotic simulation, making it practical for real-world deployment.
Problem

Research questions and friction points this paper is trying to address.

Integrates semantic segmentation with geometric reconstruction for robotics
Improves semantic segmentation accuracy using hybrid SAM2 model
Enables efficient human tracking and point cloud fusion
Innovation

Methods, ideas, or system contributions that make the work stand out.

End-to-end modular pipeline for RGB-D processing
Hybrid semantic segmentation with SAM2 and classification
Efficient point cloud fusion using semantic information
🔎 Similar Papers
No similar papers found.
💼 Related Jobs
No related jobs found.
Analog Devices
Z
Zhiwu Zheng
Analog Garage, Analog Devices, Inc., Boston, MA 02110, USA
L
Lauren Mentzer
Analog Garage, Analog Devices, Inc., Boston, MA 02110, USA
B
Berk Iskender
Analog Garage, Analog Devices, Inc., Boston, MA 02110, USA
M
Michael Price
Analog Garage, Analog Devices, Inc., Boston, MA 02110, USA
C
Colm Prendergast
Analog Garage, Analog Devices, Inc., Boston, MA 02110, USA
A
A. Cloitre
Analog Garage, Analog Devices, Inc., Boston, MA 02110, USA