Invisible Stitch: Generating Smooth 3D Scenes with Depth Inpainting

Paul Engstler,Andrea Vedaldi,Iro Laina,Christian Rupprecht
2024-05-01
Abstract:3D scene generation has quickly become a challenging new research direction, fueled by consistent improvements of 2D generative diffusion models. Most prior work in this area generates scenes by iteratively stitching newly generated frames with existing geometry. These works often depend on pre-trained monocular depth estimators to lift the generated images into 3D, fusing them with the existing scene representation. These approaches are then often evaluated via a text metric, measuring the similarity between the generated images and a given text prompt. In this work, we make two fundamental contributions to the field of 3D scene generation. First, we note that lifting images to 3D with a monocular depth estimation model is suboptimal as it ignores the geometry of the existing scene. We thus introduce a novel depth completion model, trained via teacher distillation and self-training to learn the 3D fusion process, resulting in improved geometric coherence of the scene. Second, we introduce a new benchmarking scheme for scene generation methods that is based on ground truth geometry, and thus measures the quality of the structure of the scene.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper addresses the problems in 3D scene generation, namely geometric inconsistency and inaccurate depth prediction when generating the entire scene from a single image or textual description. Existing methods rely on pre-trained monocular depth estimators, ignoring the geometric information of the existing scene. To solve this, the paper proposes two contributions: 1) a depth completion model that improves the geometric coherence of the scene through teacher distillation and self-training to learn 3D fusion process, and 2) a new benchmark based on real geometry to evaluate the quality of scene generation methods, no longer solely relying on image similarity. This approach reduces the discontinuities in the generated scene, improves consistency, and demonstrates better results than existing methods in experiments.