Downloads · 30 days
0
cy0307/b-gaussian-splatting-2d
b-gaussian-splatting-2d is a machine learning model from cy0307. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for pytorch. The card lists the license as mit.
Reconstructs a real photograph with anisotropic 2D Gaussians (with densification) — the 2D analogue of 3D Gaussian Splatting.
Downloads · 30 days
0
Access
Public
Updated Jun 28, 2026
Repo size
78.7 KB
Likes
12
Public
Click a slice to open those files.
.png53.5 KB · 62%
From the Hugging Face model README
Reconstructs a real photograph with anisotropic 2D Gaussians (with densification) — the 2D analogue of 3D Gaussian Splatting.
Trained from scratch in Ropedia Academy — an interactive, bilingual course on embodied & spatial AI. Educational model: small and quick to train; the value is the method and a reproducible pipeline, not a leaderboard score. Try it live in the Ropedia demos Space.
Adam (lr 0.02), 800 steps; 500 Gaussians, gradient-based densification (→ ~650); 64×64 target.
| metric | value | meaning |
|---|---|---|
psnr (final) | 32.45 |

import torch
g = torch.load("gaussians.pt", map_location="cpu") # dict: pos, logs, rot, col, op
# Re-create render() from the notebook (see "Reproduce") and call it on these tensors
# to reconstruct the fitted image.
Educational scale. Trained quickly on CPU on small or synthetic data, so absolute numbers are not competitive with production systems — the value is the method and a reproducible pipeline. No large-scale data, no hyperparameter sweep, and no multi-seed variance is reported. Not for production use.
Overfits a single image — it does not generalize to other images; quality is capped by the Gaussian count.
Without densification, large flat regions stay blurry; over-large σ washes the image out.
One click: open the notebook in Colab → Runtime → GPU → Run all, then run its Publish to the Hugging Face Hub cell.
From a shell:
git clone https://github.com/ChaoYue0307/ropedia-academy.git && cd ropedia-academy
pip install torch numpy matplotlib scikit-learn scikit-image gymnasium
jupyter nbconvert --to notebook --execute notebooks/training/B_gaussian_splatting_2d.ipynb --output run.ipynb
# optional: override training length, e.g. STEPS=2000 (or EPISODES=600) before running
figure.pnggaussians.ptmetrics.jsonCode & weights: MIT (this repository) — educational use encouraged.
Image: astronaut test image (NASA) — public domain, shipped with scikit-image.
If you use this model or the course materials, please cite:
@misc{ropedia_academy,
title = {Ropedia Academy: an interactive course on embodied & spatial AI},
author = {Ropedia Academy},
year = {2026},
howpublished = {\url{https://chaoyue0307.github.io/ropedia-academy/}}
}
Method / original work: Kerbl et al., 3D Gaussian Splatting for Real-Time Radiance Field Rendering, SIGGRAPH 2023.
Part of the Ropedia Academy trained-model collection. Contributions & issues welcome on GitHub.