SimpleProc
收藏资源简介:
SimpleProc是一个完全通过程序规则生成的合成数据集,用于多视图立体视觉(MVS)的训练。它使用非均匀有理B样条(NURBS)以及基本的位移和纹理模式,通过简单的规则生成有效的训练数据。在8,000张图像的规模下,该方法比从游戏和现实世界物体中手动获取的图像(相同规模)表现更优。当扩展到352,000张图像时,其性能与在超过692,000张手动获取的图像上训练的模型相当,甚至在某些基准测试中表现更优。
We explore the design space of procedural rules for multi-view stereo (MVS). We demonstrate that effective training data can be generated using SimpleProc: a fully procedural generator driven by a set of extremely simple rules, which leverages non-uniform rational B-splines (NURBS) alongside basic displacement and texture patterns. At a modest scale of 8,000 images, our method yields better performance than models trained on images manually curated from games and real-world objects at the same scale. When scaled up to 352,000 images, our method matches or even outperforms models trained on over 692,000 manually curated images across multiple benchmark datasets.
SimpleProc 数据集概述
数据集基本信息
- 数据集名称:SimpleProc
- 核心内容:用于多视图立体视觉(MVS)训练的全程序化合成数据
- 生成方法:基于简单规则,使用非均匀有理B样条、基本位移和纹理模式进行全程序化生成
- 数据规模:论文中使用了44,000个场景;生成器可支持更大规模(如8,000至352,000张图像)的数据生成
数据生成与获取
数据生成器准备
- 环境配置:需创建Conda环境(Python 3.11),并安装Infinigen的API。
- 依赖安装:需安装Blender 4.5(用于EEVEE渲染)及相关Python库(opencv-python, OpenEXR==3.3.5, matplotlib)。
- 生成方式:
- 本地生成单场景:通过指定种子和输出路径,生成Blend场景文件并进行渲染(需GPU)。
- 集群批量生成:可通过脚本批量生成多个场景。
论文数据下载
- 44,000个场景数据:可从提供的Dropbox链接下载(https://www.dropbox.com/scl/fo/snn2vebj5a84ro2m7mqou/AKCdc6-RFRMfpICQ5_PItsY?rlkey=0erbn8i2yhacy3xb54m27ber7&st=k22pyfky&dl=0)。
- 训练验证数据:需从另一Dropbox链接下载(https://www.dropbox.com/scl/fi/9zcrtpb3xdzz71s7fkgg7/rmvd_samples_eval.zip?rlkey=kgekhjpe4begrhhl4fop41bhg&st=hwdm3ctt&dl=0),用于训练监控。
- 预训练模型:可从Dropbox链接下载(https://www.dropbox.com/scl/fi/0gy1hxo41r0sudtzz4sml/checkpoints.zip?rlkey=er1h7vim2rnom2f8c580wcy5w&st=3po98loz&dl=0),包含3次不同运行的检查点。
模型训练
- 训练框架:基于MVSAnywhere的分支版本(https://github.com/mazeyu/mvsanywhere)。
- 配置要求:需在配置文件中指定数据集路径和验证数据目录。
- 硬件要求:GPU内存需48GB。
- 训练命令:提供了完整的训练脚本示例,包括批次大小、最大步数等参数。
相关资源
- 学术论文:标题为《SimpleProc: Fully Procedural Synthetic Data from Simple Rules for Multi-View Stereo》,发布于arXiv(https://arxiv.org/abs/2604.04925)。
- 作者:Zeyu Ma, Alexander Raistrick, Jia Deng。
- 代码仓库:基于Infinigen API,主仓库地址为 https://github.com/princeton-vl/SimpleProc。




