OpenMask3D: Open-Vocabulary 3D Instance Segmentation
About
We introduce the task of open-vocabulary 3D instance segmentation. Current approaches for 3D instance segmentation can typically only recognize object categories from a pre-defined closed set of classes that are annotated in the training datasets. This results in important limitations for real-world applications where one might need to perform tasks guided by novel, open-vocabulary queries related to a wide variety of objects. Recently, open-vocabulary 3D scene understanding methods have emerged to address this problem by learning queryable features for each point in the scene. While such a representation can be directly employed to perform semantic segmentation, existing methods cannot separate multiple object instances. In this work, we address this limitation, and propose OpenMask3D, which is a zero-shot approach for open-vocabulary 3D instance segmentation. Guided by predicted class-agnostic 3D instance masks, our model aggregates per-mask features via multi-view fusion of CLIP-based image embeddings. Experiments and ablation studies on ScanNet200 and Replica show that OpenMask3D outperforms other open-vocabulary methods, especially on the long-tail distribution. Qualitative experiments further showcase OpenMask3D's ability to segment object properties based on free-form queries describing geometry, affordances, and materials.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 3D Semantic Segmentation | ScanNet (val) | mIoU34 | 100 | |
| Semantic segmentation | ScanNet V2 | mIoU34 | 54 | |
| Instance Segmentation | ScanNet200 (val) | mAP@5019.9 | 53 | |
| 3D Instance Segmentation | ScanNet200 (val) | mAP15.4 | 52 | |
| 3D Instance Segmentation | ScanNet200 | mAP@0.519.9 | 29 | |
| 3D Semantic Segmentation | ScanNet200 (val) | mIoU (All Classes)10.3 | 14 | |
| Class-agnostic 3D instance segmentation | ScanNet200 (val) | AP39.7 | 12 | |
| Semantic segmentation | ScanNet200 | mIoU10.5 | 12 | |
| Open-vocabulary 3D instance segmentation | ScanNet200 v1 (val) | AP Novel15 | 7 | |
| Open-vocabulary 3D instance segmentation | ScanNet200 (val) | mAP15.4 | 7 |