Optimal Pose and Shape Estimation for Category-level 3D Object Perception
arXiv:2104.08383 · doi:10.15607/RSS.2021.XVII.025
Abstract
We consider a category-level perception problem, where one is given 3D sensor data picturing an object of a given category (e.g. a car), and has to reconstruct the pose and shape of the object despite intra-class variability (i.e. different car models have different shapes). We consider an active shape model, where -- for an object category -- we are given a library of potential CAD models describing objects in that category, and we adopt a standard formulation where pose and shape estimation are formulated as a non-convex optimization. Our first contribution is to provide the first certifiably optimal solver for pose and shape estimation. In particular, we show that rotation estimation can be decoupled from the estimation of the object translation and shape, and we demonstrate that (i) the optimal object rotation can be computed via a tight (small-size) semidefinite relaxation, and (ii) the translation and shape parameters can be computed in closed-form given the rotation. Our second contribution is to add an outlier rejection layer to our solver, hence making it robust to a large number of misdetections. Towards this goal, we wrap our optimal solver in a robust estimation scheme based on graduated non-convexity. To further enhance robustness to outliers, we also develop the first graph-theoretic formulation to prune outliers in category-level perception, which removes outliers via convex hull and maximum clique computations; the resulting approach is robust to 70%-90% outliers. Our third contribution is an extensive experimental evaluation. Besides providing an ablation study on a simulated dataset and on the PASCAL3D+ dataset, we combine our solver with a deep-learned keypoint detector, and show that the resulting approach improves over the state of the art in vehicle pose estimation in the ApolloScape datasets.
References in corpus (25)
- Very Deep Convolutional Networks for Large-Scale Image Recognition
- DeepPose: Human Pose Estimation via Deep Neural Networks
- Depth Map Prediction from a Single Image using a Multi-Scale Deep Network
- Joint Training of a Convolutional Network and a Graphical Model for Human Pose Estimation
- The ApolloScape Open Dataset for Autonomous Driving and its Application
- Graduated Non-Convexity for Robust Spatial Perception: From Non-Minimal Solvers to Global Outlier Rejection
- Scalability in Perception for Autonomous Driving: Waymo Open Dataset
- Guaranteed Outlier Removal for Point Cloud Registration with Correspondences
- Sparse Representation for 3D Shape Estimation: A Convex Relaxation Approach
- Matryoshka Networks: Predicting 3D Geometry via Nested Shape Layers
- kPAM: KeyPoint Affordances for Category-Level Robotic Manipulation
- Towards Robust Monocular Depth Estimation: Mixing Datasets for Zero-shot Cross-dataset Transfer
- PVNet: Pixel-wise Voting Network for 6DoF Pose Estimation
- TEASER: Fast and Certifiable Point Cloud Registration
- MobilePose: Real-Time Pose Estimation for Unseen Objects with Weak Shape Supervision
- TSSOS: A Moment-SOS hierarchy that exploits term sparsity
- ApolloCar3D: A Large 3D Car Instance Understanding Benchmark for Autonomous Driving
- Fully Convolutional Geometric Features for Category-level Object Alignment
- Learning Canonical Shape Space for Category-Level 6D Object Pose and Size Estimation
- Probabilistic Category-Level Pose Estimation via Segmentation and Predicted-Shape Priors
- GSNet: Joint Vehicle Pose and Shape Reconstruction with Geometrical and Scene-aware Supervision
- A Quaternion-based Certifiably Optimal Solution to the Wahba Problem with Outliers
- One Ring to Rule Them All: Certifiably Robust Geometric Perception with Outliers
- Outlier-Robust Estimation: Hardness, Minimally Tuned Algorithms, and Applications
- In Perfect Shape: Certifiably Optimal 3D Shape Reconstruction from 2D Landmarks
Cited by in corpus (4)
- Optimal and Robust Category-level Perception: Object Pose and Shape Estimation from 2D and 3D Semantic Keypoints
- A Certifiable Algorithm for Simultaneous Shape Estimation and Object Tracking
- An Inexact Projected Gradient Method with Rounding and Lifting by Nonlinear Programming for Solving Rank-One Semidefinite Relaxation of Polynomial Optimization
- Dynamical Pose Estimation