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

Open3DIS: Open-Vocabulary 3D Instance Segmentation with 2D Mask Guidance

About

We introduce Open3DIS, a novel solution designed to tackle the problem of Open-Vocabulary Instance Segmentation within 3D scenes. Objects within 3D environments exhibit diverse shapes, scales, and colors, making precise instance-level identification a challenging task. Recent advancements in Open-Vocabulary scene understanding have made significant strides in this area by employing class-agnostic 3D instance proposal networks for object localization and learning queryable features for each 3D mask. While these methods produce high-quality instance proposals, they struggle with identifying small-scale and geometrically ambiguous objects. The key idea of our method is a new module that aggregates 2D instance masks across frames and maps them to geometrically coherent point cloud regions as high-quality object proposals addressing the above limitations. These are then combined with 3D class-agnostic instance proposals to include a wide range of objects in the real world. To validate our approach, we conducted experiments on three prominent datasets, including ScanNet200, S3DIS, and Replica, demonstrating significant performance gains in segmenting objects with diverse categories over the state-of-the-art approaches.

Phuc D.A. Nguyen, Tuan Duc Ngo, Evangelos Kalogerakis, Chuang Gan, Anh Tran, Cuong Pham, Khoi Nguyen• 2023

Related benchmarks

TaskDatasetResultRank
3D Instance SegmentationScanNet200 (val)
mAP41.5
52
3D Instance SegmentationScanNet200
mAP@0.543.1
29
3D Instance SegmentationS3DIS (B8/N4)
mAP50 (Base)60.8
13
3D Instance SegmentationS3DIS B6 N6
mAP50 (Base)50
13
Class-agnostic 3D instance segmentationScanNet200 (val)
AP34.6
12
3D Open-vocabulary Instance SegmentationS3DIS 8/4 base/novel categories split
AP@5026.3
11
3D Open-vocabulary Instance SegmentationS3DIS (6/6 base/novel categories split)
AP@5029
11
3D Instance SegmentationScanNet200 v1 (test)
AP41.5
9
Class-agnostic 3D instance segmentationScanNet++ 1554 classes 1.0 (val)
AP20.7
8
3D Instance Segmentation3RScan ScanNet200→3RScan transfer (test)
AP9.5
8
Showing 10 of 18 rows

Other info

Code

Follow for update