Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Segment3D: Learning Fine-Grained Class-Agnostic 3D Segmentation without Manual Labels

About

Current 3D scene segmentation methods are heavily dependent on manually annotated 3D training datasets. Such manual annotations are labor-intensive, and often lack fine-grained details. Importantly, models trained on this data typically struggle to recognize object classes beyond the annotated classes, i.e., they do not generalize well to unseen domains and require additional domain-specific annotations. In contrast, 2D foundation models demonstrate strong generalization and impressive zero-shot abilities, inspiring us to incorporate these characteristics from 2D models into 3D models. Therefore, we explore the use of image segmentation foundation models to automatically generate training labels for 3D segmentation. We propose Segment3D, a method for class-agnostic 3D scene segmentation that produces high-quality 3D segmentation masks. It improves over existing 3D segmentation models (especially on fine-grained masks), and enables easily adding new training data to further boost the segmentation performance -- all without the need for manual training labels.

Rui Huang, Songyou Peng, Ayca Takmaz, Federico Tombari, Marc Pollefeys, Shiji Song, Gao Huang, Francis Engelmann• 2023

Related benchmarks

TaskDatasetResultRank
Class-agnostic 3D instance segmentationScanNet++ 1554 classes 1.0 (val)
AP19
8
Class-agnostic 3D instance segmentationScanNet++
AP12
7
Class-agnostic 3D instance segmentationS3DIS (test)
AP21.7
7
Class-agnostic 3D instance segmentationScanNet++ (test)
AP12
7
Class-agnostic 3D instance segmentationScanNet V2 (val)
AP28.3
7
Open-vocabulary 3D instance segmentationScanNet++ 100 classes
AP10.1
4
Showing 6 of 6 rows

Other info

Follow for update