Check out the source code on GitHub.
We provide a dataset handler for ShapeStacks using Tensorflow dataset API. It also contains example scripts for training and evaluation of the stablity prediction models.
Physical intuition is pivotal for intelligent agents to perform complex tasks. In this paper we investigate the passive acquisition of an intuitive understanding of physical principles as well as the active utilisation of this intuition in the context of generalised object stacking. To this end, we provide ShapeStacks: a simulation-based dataset featuring 20,000 stack configurations composed of a variety of elementary geometric primitives richly annotated regarding semantics and structural stability.
|Oliver Groth, Fabian Fuchs, Ingmar Posner, Andrea Vedaldi||Applied AI Lab, Visual Geometry Group||Research funded by ERC 677195-IDIU, AIMS-CDT|