Gaussian Splatting 研究项目

这个项目实现了3D和2D Gaussian Splatting的核心算法，用于神经渲染和图像重建研究。

项目结构

gaussian_splatting/
├── src/
│   ├── models/
│   │   ├── __init__.py
│   │   ├── gaussian_3d.py      # 3D Gaussian Splatting模型
│   │   ├── gaussian_2d.py      # 2D Gaussian Splatting模型
│   │   └── renderer.py         # 渲染器
│   ├── utils/
│   │   ├── __init__.py
│   │   ├── geometry.py         # 几何变换工具
│   │   ├── visualization.py    # 可视化工具
│   │   └── data_loader.py      # 数据加载器
│   └── training/
│       ├── __init__.py
│       ├── trainer.py          # 训练器
│       └── losses.py           # 损失函数
├── scripts/
│   ├── train_3d.py            # 3D模型训练脚本
│   ├── train_2d.py            # 2D模型训练脚本
│   ├── test_3d.py             # 3D模型测试脚本
│   └── test_2d.py             # 2D模型测试脚本
├── data/                      # 数据目录
├── outputs/                   # 输出目录
├── configs/                   # 配置文件
└── requirements.txt           # 依赖包

安装

pip install -r requirements.txt

使用方法

3D Gaussian Splatting

# 训练3D模型
python scripts/train_3d.py --config configs/3d_config.yaml

# 测试3D模型
python scripts/test_3d.py --model_path outputs/3d_model.pth

2D Gaussian Splatting

# 训练2D模型
python scripts/train_2d.py --config configs/2d_config.yaml

# 测试2D模型
python scripts/test_2d.py --model_path outputs/2d_model.pth

特性

• 完整的3D Gaussian Splatting实现
• 2D Gaussian Splatting扩展
• 可配置的训练参数
• 实时渲染支持
• 多种损失函数
• 可视化工具

参考文献

• 3D Gaussian Splatting for Real-Time Radiance Field Rendering
• 相关2D扩展论文