# stochasticgs

**Repository Path**: hineven/stochasticgs

## Basic Information

- **Project Name**: stochasticgs
- **Description**: 实验用仓库
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2026-03-04
- **Last Updated**: 2026-03-13

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# Stochastic 3DGS Recurrent Denoising

本仓库当前聚焦于 **kernel-predictor 时序降噪**，用于对 stochastic / real-time 3D Gaussian Splatting 渲染结果进行降噪。

当前保留的核心能力：

- `CameraTrack` 生成与管理
- `RenderedTrackData` 训练数据采集与按块加载
- 基于相机位姿与 Reversed-Z 深度的时序重投影
- kernel-predictor 训练与评估

## 当前训练入口

推荐默认训练命令（多轨流式、GroupNorm、有效 batch 16）：

`python train.py --data-dir ./training_data --streaming on --batch-size 2 --effective-batch-size 16 --num-loaders 6 --steps 50000 --precision auto --warmup-loss-mode mask --warmup-sampling mixed --warmup-frames 7 --supervised-frames 2 --sequence-length 9 --tbptt-steps 1 --norm-type groupnorm --norm-groups 8 --resume off`

单轨训练：

`python train.py --data-dir ./training_data/scene_xxx/rendered/orbit_001 --sequence-length 6 --warmup-frames 2 --sequence-stride 3 --batch-size 1`

多轨流式训练：

`python train.py --data-dir ./training_data --streaming on --sequence-length 6 --warmup-frames 2 --sequence-stride 3 --batch-size 1 --buffer-size 8 --num-loaders 4`

说明：

- `--effective-batch-size 16` 会自动将 `batch-size 2` 转成 `grad_accum_steps 8`
- 当前推荐归一化为 `--norm-type groupnorm --norm-groups 8`
- 开启新实验时建议使用 `--resume off`，避免误载入旧 checkpoint
- 若要继续旧版无归一化层权重，请显式使用 `--norm-type none`

评估：

`python test.py --data-dir ./testing_data --model-path ./model_weights/kernel_predictor_denoiser/weights.pt --save-vis ./eval_vis`

## 主要数据抽象

- `Scene`：场景与模型实例
- `SceneMeta`：场景轻量元数据
- `CameraTrack`：相机轨迹
- `RenderedTrackData`：一条轨迹的渲染结果与元数据
- `RenderedTrackBlock`：渲染数据分块缓存
- `SequenceTrackDataset` / `MultiTrackSequenceDataset`：recurrent 训练序列数据集

## 说明

- 仓库已移除普通单帧 `U-Net` 训练 / 测试支持
- `train.py` 与 `test.py` 仅接受 recurrent 工作流
- 旧版 `view_*` 单帧数据格式不再作为训练入口支持