HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

DINO in the Room: Leveraging 2D Foundation Models for 3D Segmentation

Karim Abou Zeid Kadir Yilmaz Daan de Geus Alexander Hermans David Adrian Timm Linder Bastian Leibe

DINO in the Room: Leveraging 2D Foundation Models for 3D Segmentation

Abstract

Vision foundation models (VFMs) trained on large-scale image datasets provide high-quality features that have significantly advanced 2D visual recognition. However, their potential in 3D vision remains largely untapped, despite the common availability of 2D images alongside 3D point cloud datasets. While significant research has been dedicated to 2D-3D fusion, recent state-of-the-art 3D methods predominantly focus on 3D data, leaving the integration of VFMs into 3D models underexplored. In this work, we challenge this trend by introducing DITR, a simple yet effective approach that extracts 2D foundation model features, projects them to 3D, and finally injects them into a 3D point cloud segmentation model. DITR achieves state-of-the-art results on both indoor and outdoor 3D semantic segmentation benchmarks. To enable the use of VFMs even when images are unavailable during inference, we further propose to distill 2D foundation models into a 3D backbone as a pretraining task. By initializing the 3D backbone with knowledge distilled from 2D VFMs, we create a strong basis for downstream 3D segmentation tasks, ultimately boosting performance across various datasets.

Code Repositories

VisualComputingInstitute/DITR
Official
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
3d-semantic-segmentation-on-scannet200DITR
test mIoU: 44.9
val mIoU: 41.2
3d-semantic-segmentation-on-semantickittiDITR
test mIoU: 74.4%
val mIoU: 69.0%
3d-semantic-segmentation-on-waymo-openDITR
mIoU: 73.3
lidar-semantic-segmentation-on-nuscenesDITR
test mIoU: 0.851
val mIoU: 0.842
semantic-segmentation-on-s3dis-area5DITR
mIoU: 74.1
semantic-segmentation-on-scannetDITR
test mIoU: 79.7
val mIoU: 80.5

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
DINO in the Room: Leveraging 2D Foundation Models for 3D Segmentation | Papers | HyperAI