arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.16874cs.CV

CheckerPose:利用图神经网络实现渐进式密集关键点定位的物体位姿估计

CheckerPose: Progressive Dense Keypoint Localization for Object Pose Estimation with Graph Neural Network

Stony Brook University(石溪大学)

Ruyi Lian, Haibin Ling

AI总结:

本文提出 CheckerPose,通过三维表面关键点渐进定位、二进制像素编码和图神经网络建模,提升单张 RGB 图像的 6-DoF 物体位姿估计精度,并在多个基准上达到最优。

摘要:

从单张 RGB 图像估计刚性物体的 6-DoF 位姿是一项关键且具有挑战性的任务。近期研究表明,基于密集对应的解决方案具有巨大潜力,但要实现实际部署仍需进一步改进。本文提出一种名为 CheckerPose 的新型位姿估计算法,在三个主要方面进行了改进。首先,CheckerPose 从三维物体表面密集采样三维关键点,并在二维图像中渐进式地寻找它们的二维对应点。与以往在图像空间中进行密集采样的解决方案相比,我们的策略使对应搜索能够在二维网格(即像素坐标)中进行。其次,针对我们的三维到二维对应,设计了一种紧凑的二进制编码表示来表示二维图像位置。该表示不仅允许渐进式对应细化,还将对应回归转化为更高效的分类问题。第三,采用图神经网络显式建模所采样三维关键点之间的交互,进一步提升对应关系的可靠性和准确性。这些新颖组件共同使 CheckerPose 成为一种强大的位姿估计算法。在流行的 Linemod、Linemod-O 和 YCB-V 物体位姿估计基准上进行评估时,CheckerPose 明显提升了基于对应方法的准确性,并取得了最先进的性能。代码可在 https://github.com/RuyiLian/CheckerPose 获取。

英文摘要:

Estimating the 6-DoF pose of a rigid object from a single RGB image is a crucial yet challenging task. Recent studies have shown the great potential of dense correspondence-based solutions, yet improvements are still needed to reach practical deployment. In this paper, we propose a novel pose estimation algorithm named CheckerPose, which improves on three main aspects. Firstly, CheckerPose densely samples 3D keypoints from the surface of the 3D object and finds their 2D correspondences progressively in the 2D image. Compared to previous solutions that conduct dense sampling in the image space, our strategy enables the correspondence searching in a 2D grid (i.e., pixel coordinate). Secondly, for our 3D-to-2D correspondence, we design a compact binary code representation for 2D image locations. This representation not only allows for progressive correspondence refinement but also converts the correspondence regression to a more efficient classification problem. Thirdly, we adopt a graph neural network to explicitly model the interactions among the sampled 3D keypoints, further boosting the reliability and accuracy of the correspondences. Together, these novel components make CheckerPose a strong pose estimation algorithm. When evaluated on the popular Linemod, Linemod-O, and YCB-V object pose estimation benchmarks, CheckerPose clearly boosts the accuracy of correspondence-based methods and achieves state-of-the-art performances. Code is available at https://github.com/RuyiLian/CheckerPose.

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

↑