Diverse Plausible Shape Completions from Ambiguous Depth Images

Abstract

We propose PSSNet, a network architecture for generating diverse plausible 3D reconstructions from a single 2.5D depth image. Existing methods tend to produce only small variations on a single shape, even when multiple shapes are consistent with an observation. To obtain diversity we alter a Variational Auto Encoder by providing a learned shape bounding box feature as side information during training. Since these features are known during training, we are able to add a supervised loss to the encoder and noiseless values to the decoder. To evaluate, we sample a set of completions from a network, construct a set of plausible shape matches for each test observation, and compare using our plausible diversity metric defined over sets of shapes. We perform experiments using Shapenet mugs and partially-occluded YCB objects and find that our method performs comparably in datasets with little ambiguity, and outperforms existing methods when many shapes plausibly fit an observed depth image. We demonstrate one use for PSSNet on a physical robot when grasping objects in occlusion and clutter.

References in corpus (10)

• Auto-Encoding Variational Bayes
• Learning a Probabilistic Latent Space of Object Shapes via 3D Generative-Adversarial Modeling
• NICE: Non-linear Independent Components Estimation
• Density estimation using Real NVP
• NVAE: A Deep Hierarchical Variational Autoencoder
• Deep Object Pose Estimation for Semantic Robotic Grasping of Household Objects
• MarrNet: 3D Shape Reconstruction via 2.5D Sketches
• Dense 3D Object Reconstruction from a Single Depth View
• Probabilistic Object Tracking using a Range Camera
• TEASER: Fast and Certifiable Point Cloud Registration