ADEPT Dataset
收藏资源简介:
ADEPT数据集用于模拟直觉物理中的期望违反,使用粗糙的概率对象表示。
The ADEPT dataset is designed to simulate expectation violations in intuitive physics, utilizing coarse probabilistic object representations.
ADEPT数据集概述
数据集描述
- 名称: ADEPT (Approximate Derenderer, Extended Physics, and Tracking)
- 研究领域: 直观物理学中的期望违反建模
- 作者: Kevin Smith, Lingjie Mei, Shunyu Yao, Jiajun Wu, Elizabeth S. Spelke, Joshua B. Tenenbaum, Tomer Ullman
- 发表: NeurIPS 2019, CogSci 2020
数据集生成
环境准备
-
克隆数据集仓库并创建conda环境。 bash git clone https://github.com/JerryLingjieMei/ADEPT-Dataset-Release cd ADEPT-Dataset-Release conda create -n adept-dataset python=3.10 conda activate adept-dataset
-
运行安装脚本,安装Blender 3.1.2及相关Python包。 bash bash install.sh
数据集配置
- 配置
ffmpeg路径。 - 可选:设置多机器运行环境。
ShapeNet对象渲染
- 下载ShapeNet Core V2并配置路径。
- 运行脚本将ShapeNet对象转换为
.blend文件。 bash ./blender/blender --background --python render/data/builder/collect_blend.py
数据集生成
-
生成训练集(例如1000个视频)。 bash ./blender/blender --background --python dataset/generate_train.py --end 1000
-
生成人类测试集。 bash ./blender/blender --background --python dataset/human/generate_human.py
数据集评估
评估方法
-
使用Python脚本收集人类测试集的结果,计算相对准确度。 bash python3 dataset/human/collect_results.py --summary_folder ${SUMMARY_FOLDER}
-
可选:自定义输出文件夹。
以上概述了ADEPT数据集的生成和评估流程,包括环境设置、数据集配置、对象渲染、数据集生成及评估方法。




