引入模拟飞行形状与模拟平面机械臂数据集

我们发布两个人工数据集，模拟飞行形状（Simulated Flying Shapes）和模拟平面机械臂（Simulated Planar Manipulator），用于测试视频处理系统的学习能力。特别地，该数据集旨在作为一种工具，便于轻松评估旨在编码、重建或预测视频帧序列的深度神经网络模型的合理性。每个数据集包含 90000 个视频。模拟飞行形状数据集包含显示两个相同形状（矩形、三角形和圆形）和大小的物体，其中一个物体接近其对应物的场景。模拟平面机械臂展示了一个 3 自由度平面机械臂执行拾放任务，需将大小变化的圆放置在方形平台上。与广泛使用的其他数据集（如 moving MNIST [1]、[2]）不同，所提出的两个数据集涉及目标导向任务（例如机械臂抓取物体并将其放置在平台上），而非展示随机运动。这使得我们的数据集更适合测试机器学习模型的预测能力及复杂运动的学习。本技术文档旨在介绍两个数据集的使用方法。