Home

Awesome

2D-Gaussian-Splatting-Reproduce

This repository contains the unofficial implementation of the paper "2D Gaussian Splatting for Geometrically Accurate Radiance Fields".

GVL lab, University of Southern California

Overview

<p float="left"> <img src="https://github.com/Han230104/2D-Gaussian-Splatting-Reproduce/blob/master/assets/garden-rgb.png?raw=true" width="350" /> <img src="https://github.com/Han230104/2D-Gaussian-Splatting-Reproduce/blob/master/assets/kitchen-rgb.png?raw=true" width="350" /> </p>

Rendered RGB Image: garden(left), kitchen(right)

Installation

Clone the repository

# SSH
git clone git@github.com:Han230104/2D-Gaussian-Splatting-Reproduce.git

or

# HTTPS
git clone https://github.com/Han230104/2D-Gaussian-Splatting-Reproduce.git

Create an anaconda environment

cd 2D-Gaussian-Splatting-Reproduce
conda env create --file environment.yml
conda activate 2dgs

pip install -r requirements.txt

Download dataset

Create a folder to store the dataset

mkdir datasets

MpiNeRF360

The MipNeRF360 scenes are hosted by the paper authors here.

After downloading the dataset, you should organize your data like this:

- 2D-Gaussian-Splatting-Reproduce
  - datasets
    - bicycle
    - bonsai
    - counter
    - flowers
    - garden
    - kitchen
    - room
    - stump
    - treehill

DTU

You can download the preprocessed data from here.

You also need to download the ground truth DTU point cloud.

After downloading the dataset, you should organize your data like this:

- 2D-Gaussian-Splatting-Reproduce
  - datasets
    - DTU_mask  # preprocessed data
      - scan105
      ...
    - DTU  # official data
      - Points
      - ObsMask

Training and Evaluation

Run the training and evaluation script

# Mip-NeRF 360 dataset
python run_mipnerf360.py
# DTU dataset
python run_dtu.py

Results

MpiNeRF360

You will get similar results like this on Mip-NeRF 360 dataset:

Ourdoor

ModelPSNR ↑SSIM ↑LIPPS ↓Time(min)
2DGS24.330.7090.28430.22
Ours24.250.7110.27822.27

Indoor

ModelPSNR ↑SSIM ↑LIPPS ↓Time(min)
2DGS30.390.9240.18227.16
Ours30.530.9250.17819.89

DTU

ModelChamfer distance ↓
2DGS0.80
Ours0.77

Acknowledgements

This project is built upon 2DGS and 3DGS. We also borrow some code from gaussian-opacity-fields. We thank all the authors for their great work and repos.