arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2102.12145cs.CVcs.RO

GDRNPP:一种几何引导的全学习式物体位姿估计器

GDRNPP: A Geometry-guided and Fully Learning-based Object Pose Estimator

Tsinghua University(清华大学);BNRist(北京国家信息科学技术研究院);China University of Geosciences Beijing(中国地质大学(北京))

Xingyu Liu, Ruida Zhang, Chenyangguang Zhang, Gu Wang, Jiwen Tang, Zhigang Li, Xiangyang Ji

AI总结:

针对刚性物体6D位姿估计中传统技术集成导致的非端到端、速度慢问题,提出全学习的GDRNPP流水线,含几何引导直接回归网络与可微优化模块,连续两年登顶BOP挑战赛,精度速度均超依赖传统技术的方法。

摘要:

刚性物体的6D位姿估计是计算机视觉中一项长期存在且具有挑战性的任务。近年来,深度学习的兴起展现了卷积神经网络(CNNs)预测可靠6D位姿的潜力。鉴于直接位姿回归网络目前表现出次优性能,大多数方法仍在不同程度上依赖传统技术。例如,性能顶尖的方法通常采用间接策略:首先建立2D-3D或3D-3D对应关系,随后应用基于RANSAC的PnP或Kabsch算法,还会进一步使用ICP进行优化。尽管性能有所提升,但传统技术的集成使得网络耗时较长且无法端到端训练。与这些方法不同,本文提出了一种全学习式的物体位姿估计器。在这项工作中,我们首先对直接和间接方法进行了深入研究,提出了一种简单而有效的几何引导直接回归网络(GDRN),以端到端的方式从单目图像中学习6D位姿。随后,我们引入了一个几何引导的位姿优化模块,在有额外深度数据可用时提升位姿精度。在预测的坐标图引导下,我们构建了一个端到端可微的架构,该架构在观测到的RGB-D图像与渲染的RGB-D图像之间建立鲁棒且准确的3D-3D对应关系,以优化位姿。我们增强后的位姿估计流水线GDRNPP(GDRN Plus Plus)连续两年登顶BOP挑战赛排行榜,成为首个在精度和速度上均超越所有依赖传统技术的现有方法的方案。代码和模型可在https://github.com/shanice-l/gdrnpp_bop2022 获取。

英文摘要:

6D pose estimation of rigid objects is a long-standing and challenging task in computer vision. Recently, the emergence of deep learning reveals the potential of Convolutional Neural Networks (CNNs) to predict reliable 6D poses. Given that direct pose regression networks currently exhibit suboptimal performance, most methods still resort to traditional techniques to varying degrees. For example, top-performing methods often adopt an indirect strategy by first establishing 2D-3D or 3D-3D correspondences followed by applying the RANSAC-based PnP or Kabsch algorithms, and further employing ICP for refinement. Despite the performance enhancement, the integration of traditional techniques makes the networks time-consuming and not end-to-end trainable. Orthogonal to them, this paper introduces a fully learning-based object pose estimator. In this work, we first perform an in-depth investigation of both direct and indirect methods and propose a simple yet effective Geometry-guided Direct Regression Network (GDRN) to learn the 6D pose from monocular images in an end-to-end manner. Afterwards, we introduce a geometry-guided pose refinement module, enhancing pose accuracy when extra depth data is available. Guided by the predicted coordinate map, we build an end-to-end differentiable architecture that establishes robust and accurate 3D-3D correspondences between the observed and rendered RGB-D images to refine the pose. Our enhanced pose estimation pipeline GDRNPP (GDRN Plus Plus) conquered the leaderboard of the BOP Challenge for two consecutive years, becoming the first to surpass all prior methods that relied on traditional techniques in both accuracy and speed. The code and models are available at https://github.com/shanice-l/gdrnpp_bop2022.

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

↑