Image2Splat Dataset
收藏资源简介:
Image2Splat是一个从单张图像生成200个视图的3D高斯溅射训练数据集的工具。它基于Microsoft TRELLIS.2和TencentARC的Pixal3D,通过5层采样器系统、探针模式、自动裁剪和AuraSR超分辨率预处理,为每个输入图像生成包含200张3000×3000 PNG渲染图像、相机内外参(Nerfstudio和COLMAP格式)的数据集,适用于PostShot、LichtFeld、gsplat等3D高斯溅射训练器。
Image2Splat is a tool for generating 3D Gaussian splatting training datasets with 200 views from a single input image. Built upon Microsoft TRELLIS.2 and Pixal3D from TencentARC, it generates datasets for each input image via a 5-layer sampler system, probe mode, automatic cropping, and AuraSR super-resolution preprocessing. Each dataset includes 200 rendered PNG images at 3000×3000 resolution, alongside camera intrinsic and extrinsic parameters in Nerfstudio and COLMAP formats, and is compatible with 3D Gaussian splatting trainers such as PostShot, LichtFeld, and gsplat.
数据集概述:Image2Splat
Image2Splat 是一个将单张物体图片转换为高质量多视角训练数据集的工具,专为 3D Gaussian Splatting(3DGS)模型训练而设计。它基于 Microsoft TRELLIS.2 和 TencentARC 的 Pixal3D 模型,并扩展了多项功能,使用户能够为不同类型的资产精细化控制生成美学效果。
核心功能与创新
- 5级采样器系统: 提供五种经过验证的配置(Tiers),分别对应不同的资产类型(如有机/石材/风化物、硬表面/机械、最大雕刻细节),作为“创意旋钮”而非固定默认值。各等级参数如下:
| 等级 | 步数 (Steps) | 形状引导 (Shape guidance) | 纹理引导 (Tex guidance) | 最佳适用场景 |
|---|---|---|---|---|
| 1 — 默认 (Default) | 12 | 7.5 | 1.1 | 安全生产基线 |
| 2 — 微妙 (Subtle) | 13 | 7.6 | 1.2 | 温和凸起,保守效果 |
| 3 — 均衡 (Balanced) | 14 | 7.7 | 1.25 | 数学中点 |
| 4 — 精致 (Refined) | 15 | 7.6 | 1.2 | 硬表面 + 有机雕刻细节 |
| 5 — 雕刻 (Sculpted) | 15 | 8.0 | 1.5 | 最大细节,硬表面,机械 |
- 探测模式 (Probe Mode): 在提交完整 200 视图渲染前,对单张图片以所有 5 个等级和种子进行单一视图(第 129 帧)快速比较,帮助用户选择最佳美学效果。
- 种子量化器 (Seed-quantifier): 在探测模式下,可指定随机种子数量或固定种子列表,以观察同一等级下不同种子的生成变化。
- 自动裁剪预处理 (AutoCrop Preprocessor): 自动识别并紧贴裁剪输入图像中的主体,去除空白背景,使主体在 Pixal3D 的 1024 像素图像条件中占据更大比例,提升有效像素密度。
- AuraSR 生成式放大预处理: 使用 GAN 网络将软/低分辨率/压缩的源图像进行 4 倍超分辨率放大,补充微细节,再送入后续流程。
- 热文件夹守护进程 (Hot-folder Daemon): 监视指定文件夹,自动处理放入其中的图片,并支持自动重启、故障隔离和断电续跑。
- Windows BAT 文件: 提供可直接运行的 BAT 文件,用于启动守护进程、自动裁剪和放大功能,简化 Windows 用户操作。
输出数据
为每张输入图片生成一个独立的训练数据集,包含:
datasets/<image_slug>/USER_Alt2_200v_3000px/ images/ ← 200 张 PNG 渲染图,分辨率 3000×3000,沿轨道轨迹拍摄 000.png ... 199.png transforms.json ← Nerfstudio 格式的相机内参和外参 cameras.txt ← COLMAP 格式的外参 images.txt points3D.txt
注意: 输出的是训练数据集,而非完成训练的 3D Gaussian Splatting 模型。用户需要将数据集导入 PostShot、LichtFeld、gsplat 等训练器中进行训练。
工作流模式
- 单等级生产模式: 单图片 → 选择等级(1-5)→ 生成 200 视图 → 训练 Splat。
- 探测后生产模式: 单图片 → 探测模式(等级6)比较 → 选择最佳等级和种子 → 生成 200 视图 → 训练 Splat。
- 多等级生产模式: 单图片 → 在不同等级下多次生成 200 视图 → 在 Photoshop 中挑选最佳视图 → 训练出“最佳组合”的 Splat。
- 完整预处理链: 图片 → 自动裁剪 → AuraSR 放大 → 选择等级生成 200 视图 → 训练 Splat。
已知限制与未来规划
- GPU 锁定: 需要支持 CUDA 的 NVIDIA GPU(≥24 GB VRAM),已验证在 RTX 5090/4090/A6000/H100 上运行,Apple Silicon 或 CPU 路径未经测试。
- 单资产方向: 假定输入图像主轴与资产的垂直方向一致,对于特殊姿势的图片可能需要预先旋转。
- 探测模式单视图限制: 目前探测模式仅限于单视图(第129帧),未来可能扩展至多视图探测。
技术架构
输入图片 → [可选:自动裁剪] → [可选:AuraSR 放大] → 热文件夹守护进程 → 预处理(背景移除) → 相机估计 → Pixal3D 生成 → 网格提取 → 多视角渲染(200 视图,3000px,HDRI 照明) → 后处理(锐化+噪点) → COLMAP / Nerfstudio 数据集导出 → 最终数据集
代码仓库布局
src/:包含流水线的核心 Python 源代码(骨干网络、相机、渲染、后处理等)。scripts/:包含各种运行脚本和工具(守护进程、裁剪、探测验证等)。upscale/:AuraSR 放大脚本。bat/:Windows 系统用的 BAT 文件。
许可与致谢
- 许可协议: MIT
- 致谢: 项目基于 Microsoft TRELLIS.2、TencentARC Pixal3D、valeoai/MoGe、BiRefNet、nvdiffrast、fal/AuraSR 和 CuMesh 等开源项目。




