A Unified Framework for 3D Scene Understanding

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Xu, Wei, Shi, Chunsheng, Tu, Sifan, Zhou, Xin, Liang, Dingkang, Bai, Xiang
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866915036804415488
author Xu, Wei
Shi, Chunsheng
Tu, Sifan
Zhou, Xin
Liang, Dingkang
Bai, Xiang
author_facet Xu, Wei
Shi, Chunsheng
Tu, Sifan
Zhou, Xin
Liang, Dingkang
Bai, Xiang
contents We propose UniSeg3D, a unified 3D scene understanding framework that achieves panoptic, semantic, instance, interactive, referring, and open-vocabulary segmentation tasks within a single model. Most previous 3D segmentation approaches are typically tailored to a specific task, limiting their understanding of 3D scenes to a task-specific perspective. In contrast, the proposed method unifies six tasks into unified representations processed by the same Transformer. It facilitates inter-task knowledge sharing, thereby promoting comprehensive 3D scene understanding. To take advantage of multi-task unification, we enhance performance by establishing explicit inter-task associations. Specifically, we design knowledge distillation and contrastive learning methods to transfer task-specific knowledge across different tasks. Experiments on three benchmarks, including ScanNet20, ScanRefer, and ScanNet200, demonstrate that the UniSeg3D consistently outperforms current SOTA methods, even those specialized for individual tasks. We hope UniSeg3D can serve as a solid unified baseline and inspire future work. Code and models are available at https://github.com/dk-liang/UniSeg3D.
format Preprint
id arxiv_https___arxiv_org_abs_2407_03263
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Unified Framework for 3D Scene Understanding
Xu, Wei
Shi, Chunsheng
Tu, Sifan
Zhou, Xin
Liang, Dingkang
Bai, Xiang
Computer Vision and Pattern Recognition
We propose UniSeg3D, a unified 3D scene understanding framework that achieves panoptic, semantic, instance, interactive, referring, and open-vocabulary segmentation tasks within a single model. Most previous 3D segmentation approaches are typically tailored to a specific task, limiting their understanding of 3D scenes to a task-specific perspective. In contrast, the proposed method unifies six tasks into unified representations processed by the same Transformer. It facilitates inter-task knowledge sharing, thereby promoting comprehensive 3D scene understanding. To take advantage of multi-task unification, we enhance performance by establishing explicit inter-task associations. Specifically, we design knowledge distillation and contrastive learning methods to transfer task-specific knowledge across different tasks. Experiments on three benchmarks, including ScanNet20, ScanRefer, and ScanNet200, demonstrate that the UniSeg3D consistently outperforms current SOTA methods, even those specialized for individual tasks. We hope UniSeg3D can serve as a solid unified baseline and inspire future work. Code and models are available at https://github.com/dk-liang/UniSeg3D.
title A Unified Framework for 3D Scene Understanding
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.03263