3D Wheat View Synthesis Dataset
收藏数据集概述
数据集描述
该数据集用于高保真小麦植物三维重建,使用3D高斯喷射和神经辐射场技术。数据集包含20种不同的小麦植物,每种植物在6个不同的时间帧被成像,但由于设备故障,总共只捕获了112个实例。每个捕获的实例都生成了相机在三维空间中的变换信息,适用于当前的视图合成模型,如NeRF和3DGS。
数据集访问
数据集可通过以下链接访问:数据集链接
数据集结构
数据集的结构如下:
+-- Scene (plant) #1
| +-- Date #1
| | +-- colmap
| | +-- exports
| | +-- gaussian-splatting
| | +-- images
| | +-- nerf
| | +-- transforms
| +-- Date #2
| ...
| +-- Date #6
+-- Scene (plant) #2
...
+-- Scene (plant) #20
每个顶层目录存储单个小麦植物的所有信息,每个日期目录包含在该特定日期收集的所有数据。重要的子目录包括:
- Images: 存储所有捕获的RGB图像以及生成的深度图、分割图像和掩码。
- Transforms: 包含每个图像的所有变换信息,包括原始、调整、深度、分割和未失真变换。
- Gaussian-Splatting: 存储所有训练的3D高斯喷射模型数据,包括评估指标和导出的.splat文件。
- NeRF: 存储所有训练的NeRF模型数据,包括评估指标。
- Colmap: 包含在捆绑调整过程中生成的所有SfM数据,用于优化变换。
- Exports: 存储每个训练模型的所有网格、点云、渲染图像和渲染视频。
数据集使用
数据集兼容当前的视图合成模型,如NeRF和3DGS。可以使用NeRFStudio或其他遵循标准数据输入约定的模型进行训练。
运行模型
提供了一个run_models.py脚本,可以用于查看预训练模型、训练新的视图合成数据、评估和导出图像/视频。
示例
提供了两个示例配置文件train_nerf.txt和train_3dgs.txt,可以通过以下命令运行:
bash run_models.py --config train_3dgs.txt run_models.py --config train_nerf.txt
自定义数据训练
可以训练自己的数据,只要数据具有以下目录结构:
+-- Scene
| +-- Date
| | +-- images
| | | +-- image_type
| | +-- transforms
| | | +-- transform_type
评估指标
评估指标包括由NeRFStudio评估脚本生成的nerfstudio_eval.json和由masked_psnr_calculator.py脚本生成的masked_eval.json。




