End-to-End Learnable Geometric Vision by Backpropagating PnP Optimization

Abstract

Deep networks excel in learning patterns from large amounts of data. On the other hand, many geometric vision tasks are specified as optimization problems. To seamlessly combine deep learning and geometric vision, it is vital to perform learning and geometric optimization end-to-end. Towards this aim, we present BPnP, a novel network module that backpropagates gradients through a Perspective-n-Points (PnP) solver to guide parameter updates of a neural network. Based on implicit differentiation, we show that the gradients of a "self-contained" PnP solver can be derived accurately and efficiently, as if the optimizer block were a differentiable function. We validate BPnP by incorporating it in a deep model that can learn camera intrinsics, camera extrinsics (poses) and 3D structure from training datasets. Further, we develop an end-to-end trainable pipeline for object pose estimation, which achieves greater accuracy by combining feature-based heatmap losses with 2D-3D reprojection errors. Since our approach can be extended to other optimization problems, our work helps to pave the way to perform learnable geometric vision in a principled manner. Our PyTorch implementation of BPnP is available on http://github.com/BoChenYS/BPnP.

arXiv commentCVPR 2020

References in corpus (7)

• Depth Map Prediction from a Single Image using a Multi-Scale Deep Network
• Learning Depth from Single Monocular Images Using Deep Convolutional Neural Fields
• DeMoN: Depth and Motion Network for Learning Monocular Stereo
• End-to-End Learning of Geometry and Context for Deep Stereo Regression
• Numerical Coordinate Regression with Convolutional Neural Networks
• Discovery of Latent 3D Keypoints via End-to-end Geometric Reasoning
• On Differentiating Parameterized Argmin and Argmax Problems with Application to Bi-level Optimization