Frankenstein: Generating Semantic-Compositional 3D Scenes in One Tri-Plane

Han Yan,Yang Li,Zhennan Wu,Shenzhou Chen,Weixuan Sun,Taizhang Shang,Weizhe Liu,Tian Chen,Xiaqiang Dai,Chao Ma,Hongdong Li,Pan Ji
DOI: https://doi.org/10.1145/3680528.3687672
2024-08-31
Abstract:We present Frankenstein, a diffusion-based framework that can generate semantic-compositional 3D scenes in a single pass. Unlike existing methods that output a single, unified 3D shape, Frankenstein simultaneously generates multiple separated shapes, each corresponding to a semantically meaningful part. The 3D scene information is encoded in one single tri-plane tensor, from which multiple Singed Distance Function (SDF) fields can be decoded to represent the compositional shapes. During training, an auto-encoder compresses tri-planes into a latent space, and then the denoising diffusion process is employed to approximate the distribution of the compositional scenes. Frankenstein demonstrates promising results in generating room interiors as well as human avatars with automatically separated parts. The generated scenes facilitate many downstream applications, such as part-wise re-texturing, object rearrangement in the room or avatar cloth re-targeting. Our project page is available at: <a class="link-external link-https" href="https://wolfball.github.io/frankenstein/" rel="external noopener nofollow">this https URL</a>.
Computer Vision and Pattern Recognition,Artificial Intelligence,Graphics
What problem does this paper attempt to address?
The problem this paper attempts to address is the generation of 3D scenes with semantic compositionality. Specifically, existing 3D generation models typically output single, unified 3D shapes, where the semantic information within these shapes is intertwined with other attributes, making it difficult to separate. This results in the generated 3D assets being unusable directly in downstream applications (such as decomposing generated vehicle models into the body and rollable wheels in video games, or segmenting the body, limbs, hair, and clothing in 3D digital portraits). To tackle this challenge, the paper proposes the Frankenstein framework, which aims to directly generate 3D scenes containing multiple independent semantic components. Each component has a complete shape and can undergo operations such as partial texture resetting, rearrangement of objects within a room, or redirection of clothing in digital portraits. Frankenstein encodes 3D scene information through a tri-plane tensor and uses a denoising diffusion model to generate these scenes. The main challenges include: 1. The need for a universal 3D representation method that can model the complete shapes of multiple semantic components simultaneously. 2. Modeling the relationships between different semantic parts is very complex, requiring the relative positions between parts to be semantically and physically reasonable, such as avoiding penetration phenomena. Frankenstein addresses these issues through the following steps: 1. **Tri-plane fitting**: Converting training scenes into tri-plane tensors, implicitly encoding the compositional shape information and spatial relationships between components. 2. **Variational Autoencoder (VAE) training**: Compressing the tri-plane into a more compact latent tri-plane space, significantly improving computational efficiency. 3. **Conditional denoising**: Using a diffusion model to approximate the distribution of the latent tri-plane, thereby generating 3D scenes with semantic compositionality. The paper demonstrates the effectiveness of Frankenstein in generating interior rooms and compositional portraits, showing that the generated scenes not only exhibit excellent overall quality but also have advantages in the diversity of generated shapes. The generated scenes can support various downstream applications, such as partial texture resetting, rearrangement of objects within a room, or redirection of clothing in portraits.