DiffusionGS / README.md
nielsr's picture
nielsr HF Staff
Improve model card for DiffusionGS with metadata, links, and usage example
8a0fc31 verified
|
raw
history blame
3.37 kB
metadata
license: apache-2.0
pipeline_tag: image-to-3d

Baking Gaussian Splatting into Diffusion Denoiser for Fast and Scalable Single-stage Image-to-3D Generation and Reconstruction

This repository presents DiffusionGS, a novel single-stage 3D diffusion model for object generation and scene reconstruction from a single view, as described in the paper: Baking Gaussian Splatting into Diffusion Denoiser for Fast and Scalable Single-stage Image-to-3D Generation and Reconstruction

DiffusionGS directly outputs 3D Gaussian point clouds at each timestep to enforce view consistency and allows the model to generate robustly given prompt views of any direction, beyond object-centric inputs. It boasts over 5x faster speed (~6s on an A100 GPU) compared to state-of-the-art methods.

Visual Examples

Object Generation Gaussian Splatting Object Real Image Reconstruction In the Wild Scene

Scene Diffusion 2 Scene Diffusion 1 Flux 1 Green Man

Plaza Scene Town Scene

Quick Demo

For object-centric image-to-3D generation model, we provide a single-line script to use the code:

python run.py

This code will automatically download the model checkpoints and config files from HuggingFace. Or you can manually download it from this link and set it to local dir.

For detailed instructions on installation, data preparation, evaluation, and training, please refer to the official GitHub repository.

Citation

If you find our work useful, please consider citing our paper:

@inproceedings{diffusiongs,
  title={Baking Gaussian Splatting into Diffusion Denoiser for Fast and Scalable Single-stage Image-to-3D Generation and Reconstruction},
  author={Yuanhao Cai and He Zhang and Kai Zhang and Yixun Liang and Mengwei Ren and Fujun Luan and Qing Liu and Soo Ye Kim and Jianming Zhang and Zhifei Zhang and Yuqian Zhou and Yulun Zhang and Xiaokang Yang and Zhe Lin and Alan Yuille},
  booktitle={ICCV},
  year={2025}
}