Gaussian Grouping: Segment and Edit Anything in 3D Scenes

Teaser

Gaussian Grouping extends Gaussian Splatting to jointly reconstruct and segment anything in open-world 3D scenes via lifting 2D SAM. It also efficiently supports versatile 3D scene editing tasks.

Introduction

The recent Gaussian Splatting achieves high-quality and real-time novel-view synthesis of the 3D scenes. However, it is solely concentrated on the appearance and geometry modeling, while lacking in fine-grained object-level scene understanding.

To address this issue, we propose Gaussian Grouping, which extends Gaussian Splatting to jointly reconstruct and segment anything in open-world 3D scenes. We augment each Gaussian with a compact Identity Encoding, allowing the Gaussians to be grouped according to their object instance or stuff membership in the 3D scene. Instead of resorting to expensive 3D labels, we supervise the Identity Encodings during the differentiable rendering by leveraging the 2D mask predictions by Segment Anything Model (SAM), along with introduced 3D spatial consistency regularization. Compared to the implicit NeRF representation, we show that the discrete and grouped 3D Gaussians can reconstruct, segment and edit anything in 3D with high visual quality, fine granularity and efficiency.

Based on Gaussian Grouping, we further propose a local Gaussian Editing scheme, which shows efficacy in versatile scene editing applications, including 3D object removal, inpainting, colorization, style transfer and scene recomposition.

Description of Image

Application Overview

Local Gaussian Editing scheme: Grouped Gaussians after training. Each group represents a specific instance / stuff of the 3D scene and can be fully decoupled.

Description of Image

Visualization

3D Object Removal

Our Gaussian Grouping can remove the large-scale objects on the Tanks & Temples dataset, from the whole 3D scene with greatly reduced artifacts.

3D Object Inpainting

Comparison on 3D object inpainting cases, where SPIn-NeRF requires 5h training while our method with better inpainting quality only needs 1 hour training and 20 minutes tuning.

3D Object Style Transfer

Comparison on 3D object style transfer cases, Our Gaussian Grouping produces more coherent and natural transfer results across views, with faithfully preserved background.

3D Open-world Segmentation

Our Gaussian Grouping approach jointly reconstructs and segments anything in full open-world 3D scenes. The masks predicted by Gaussian Grouping contains much sharp and accurate boundary than LERF.

3D Multi-Object Editing

Our Gaussian Grouping approach jointly reconstructs and segments anything in full open-world 3D scenes. Then we concurrently perform 3D object editing for several objects.

Related Links

Thanks a lot for the previous excellent work such as SAM, Gaussian Splatting and DEVA.

BibTeX

@article{gaussian_grouping,
      title={Gaussian Grouping: Segment and Edit Anything in 3D Scenes},
      author={Ye, Mingqiao and Danelljan, Martin and Yu, Fisher and Ke, Lei},
      journal={arXiv preprint arXiv:2312.00732},
      year={2023}
    }