Zero-Shot Multi-Object Scene Completion

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Iwase, Shun, Liu, Katherine, Guizilini, Vitor, Gaidon, Adrien, Kitani, Kris, Ambrus, Rares, Zakharov, Sergey
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909301120958464
author Iwase, Shun
Liu, Katherine
Guizilini, Vitor
Gaidon, Adrien
Kitani, Kris
Ambrus, Rares
Zakharov, Sergey
author_facet Iwase, Shun
Liu, Katherine
Guizilini, Vitor
Gaidon, Adrien
Kitani, Kris
Ambrus, Rares
Zakharov, Sergey
contents We present a 3D scene completion method that recovers the complete geometry of multiple unseen objects in complex scenes from a single RGB-D image. Despite notable advancements in single-object 3D shape completion, high-quality reconstructions in highly cluttered real-world multi-object scenes remains a challenge. To address this issue, we propose OctMAE, an architecture that leverages an Octree U-Net and a latent 3D MAE to achieve high-quality and near real-time multi-object scene completion through both local and global geometric reasoning. Because a naive 3D MAE can be computationally intractable and memory intensive even in the latent space, we introduce a novel occlusion masking strategy and adopt 3D rotary embeddings, which significantly improves the runtime and scene completion quality. To generalize to a wide range of objects in diverse scenes, we create a large-scale photorealistic dataset, featuring a diverse set of 12K 3D object models from the Objaverse dataset which are rendered in multi-object scenes with physics-based positioning. Our method outperforms the current state-of-the-art on both synthetic and real-world datasets and demonstrates a strong zero-shot capability.
format Preprint
id arxiv_https___arxiv_org_abs_2403_14628
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Zero-Shot Multi-Object Scene Completion
Iwase, Shun
Liu, Katherine
Guizilini, Vitor
Gaidon, Adrien
Kitani, Kris
Ambrus, Rares
Zakharov, Sergey
Computer Vision and Pattern Recognition
We present a 3D scene completion method that recovers the complete geometry of multiple unseen objects in complex scenes from a single RGB-D image. Despite notable advancements in single-object 3D shape completion, high-quality reconstructions in highly cluttered real-world multi-object scenes remains a challenge. To address this issue, we propose OctMAE, an architecture that leverages an Octree U-Net and a latent 3D MAE to achieve high-quality and near real-time multi-object scene completion through both local and global geometric reasoning. Because a naive 3D MAE can be computationally intractable and memory intensive even in the latent space, we introduce a novel occlusion masking strategy and adopt 3D rotary embeddings, which significantly improves the runtime and scene completion quality. To generalize to a wide range of objects in diverse scenes, we create a large-scale photorealistic dataset, featuring a diverse set of 12K 3D object models from the Objaverse dataset which are rendered in multi-object scenes with physics-based positioning. Our method outperforms the current state-of-the-art on both synthetic and real-world datasets and demonstrates a strong zero-shot capability.
title Zero-Shot Multi-Object Scene Completion
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.14628