arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2303.12787cs.CV

EPro-PnP:用于单目物体位姿估计的广义端到端概率 Perspective-n-Points

EPro-PnP: Generalized End-to-End Probabilistic Perspective-n-Points for Monocular Object Pose Estimation

Stanford University(斯坦福大学);Tongji University(同济大学);Amazon.com Inc(亚马逊公司);Alibaba Group (U.S.)(阿里巴巴集团(美国));Fudan University(复旦大学);Alibaba Group(阿里巴巴集团)

Hansheng Chen, Wei Tian, Pichao Wang, Fan Wang, Lu Xiong, Hao Li

AI总结:

本文提出概率 PnP 层 EPro-PnP,通过在 SE(3) 上预测可微位姿分布并用 KL 散度学习 2D-3D 对应,提升了 LineMOD 与 nuScenes 上的位姿估计性能。

摘要:

通过 Perspective-n-Point(PnP,透视n点)从单张 RGB 图像中定位 3D 物体是计算机视觉中的一个长期问题。在端到端深度学习的推动下,近期研究建议将 PnP 解释为一个可微层,从而通过反向传播位姿损失的梯度来部分学习 2D-3D 点对应关系。然而,从头学习完整对应关系极具挑战性,尤其是在存在模糊位姿解的情况下:此时全局最优位姿在理论上对点不可微。本文提出 EPro-PnP,这是一个用于通用端到端位姿估计的概率 PnP 层,可在 SE(3) 流形上输出具有可微概率密度的位姿分布。2D-3D 坐标及相应权重被视为中间变量,通过最小化预测位姿分布与目标位姿分布之间的 KL 散度进行学习。其底层原理推广了以往方法,并类似于注意力机制。EPro-PnP 可以增强现有对应网络,缩小基于 PnP 的方法与 LineMOD 6DoF 位姿估计基准上特定任务领先方法之间的差距。此外,EPro-PnP 有助于探索网络设计的新可能性:我们展示了一种新型可变形对应网络,在 nuScenes 3D 物体检测基准上取得了最先进的位姿精度。代码位于 https://github.com/tjiiv-cprg/EPro-PnP-v2。

英文摘要:

Locating 3D objects from a single RGB image via Perspective-n-Point (PnP) is a long-standing problem in computer vision. Driven by end-to-end deep learning, recent studies suggest interpreting PnP as a differentiable layer, allowing for partial learning of 2D-3D point correspondences by backpropagating the gradients of pose loss. Yet, learning the entire correspondences from scratch is highly challenging, particularly for ambiguous pose solutions, where the globally optimal pose is theoretically non-differentiable w.r.t. the points. In this paper, we propose the EPro-PnP, a probabilistic PnP layer for general end-to-end pose estimation, which outputs a distribution of pose with differentiable probability density on the SE(3) manifold. The 2D-3D coordinates and corresponding weights are treated as intermediate variables learned by minimizing the KL divergence between the predicted and target pose distribution. The underlying principle generalizes previous approaches, and resembles the attention mechanism. EPro-PnP can enhance existing correspondence networks, closing the gap between PnP-based method and the task-specific leaders on the LineMOD 6DoF pose estimation benchmark. Furthermore, EPro-PnP helps to explore new possibilities of network design, as we demonstrate a novel deformable correspondence network with the state-of-the-art pose accuracy on the nuScenes 3D object detection benchmark. Our code is available at https://github.com/tjiiv-cprg/EPro-PnP-v2.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑