Orienting Novel 3D Objects Using Self-Supervised Learning of Rotation Transforms

Abstract

Orienting objects is a critical component in the automation of many packing and assembly tasks. We present an algorithm to orient novel objects given a depth image of the object in its current and desired orientation. We formulate a self-supervised objective for this problem and train a deep neural network to estimate the 3D rotation as parameterized by a quaternion, between these current and desired depth images. We then use the trained network in a proportional controller to re-orient objects based on the estimated rotation between the two depth images. Results suggest that in simulation we can rotate unseen objects with unknown geometries by up to 30° with a median angle error of 1.47° over 100 random initial/desired orientations each for 22 novel objects. Experiments on physical objects suggest that the controller can achieve a median angle error of 4.2° over 10 random initial/desired orientations each for 5 objects.

References in corpus (5)

• Discovery of Latent 3D Keypoints via End-to-end Geometric Reasoning
• Deep-6DPose: Recovering 6D Object Pose from a Single RGB Image
• Mechanical Search: Multi-Step Retrieval of a Target Object Occluded by Clutter
• Pose from Shape: Deep Pose Estimation for Arbitrary 3D Objects
• Robust 6D Object Pose Estimation by Learning RGB-D Features