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

NeRF-Pose:一种用于弱监督 6D 物体姿态估计的先重建后回归方法

NeRF-Pose: A First-Reconstruct-Then-Regress Approach for Weakly-supervised 6D Object Pose Estimation

National University of Defense Technology(国防科技大学);Technical University of Munich(慕尼黑工业大学);Siemens AG(西门子股份公司)

Fu Li, Hao Yu, Ivan Shugurov, Benjamin Busam, Shaowu Yang, Slobodan Ilic

AI总结:

提出弱监督重建流程 NeRF-Pose,仅需 2D 分割和相对相机姿态,通过先重建后回归的思想结合 NeRF 与 PnP+RANSAC 实现 6D 物体姿态估计,在多个数据集上达到先进精度。

摘要:

单目图像中 3D 物体的姿态估计是计算机视觉中一个基础且长期存在的问题。现有的 6D 姿态估计深度学习方法通常依赖于 3D 物体模型和 6D 姿态标注的可用性假设。然而,在真实数据中精确标注 6D 姿态既复杂、耗时又难以扩展,而合成数据虽能很好地扩展却缺乏真实感。为避免这些问题,我们提出了一种基于重建的弱监督流程,名为 NeRF-Pose,它在训练期间仅需 2D 物体分割和已知的相对相机姿态。遵循先重建后回归的思想,我们首先以隐式神经表示的形式从多视角重建物体。然后,我们训练一个姿态回归网络,以预测图像与重建模型之间的逐像素 2D-3D 对应关系。在推理阶段,该方法仅需单张图像作为输入。使用一种基于 NeRF 的 PnP+RANSAC 算法从预测的对应关系中估计稳定且准确的姿态。在 LineMod 和 LineMod-Occlusion 上的实验表明,尽管仅使用弱标签进行训练,所提方法与最佳的 6D 姿态估计方法相比仍具有最先进的精度。此外,我们用更多真实训练图像扩展了 Homebrewed DB 数据集以支持弱监督任务,并在该数据集上取得了极具竞争力的结果。扩展的数据集和代码即将发布。

英文摘要:

Pose estimation of 3D objects in monocular images is a fundamental and long-standing problem in computer vision. Existing deep learning approaches for 6D pose estimation typically rely on the assumption of availability of 3D object models and 6D pose annotations. However, precise annotation of 6D poses in real data is intricate, time-consuming and not scalable, while synthetic data scales well but lacks realism. To avoid these problems, we present a weakly-supervised reconstruction-based pipeline, named NeRF-Pose, which needs only 2D object segmentation and known relative camera poses during training. Following the first-reconstruct-then-regress idea, we first reconstruct the objects from multiple views in the form of an implicit neural representation. Then, we train a pose regression network to predict pixel-wise 2D-3D correspondences between images and the reconstructed model. At inference, the approach only needs a single image as input. A NeRF-enabled PnP+RANSAC algorithm is used to estimate stable and accurate pose from the predicted correspondences. Experiments on LineMod and LineMod-Occlusion show that the proposed method has state-of-the-art accuracy in comparison to the best 6D pose estimation methods in spite of being trained only with weak labels. Besides, we extend the Homebrewed DB dataset with more real training images to support the weakly supervised task and achieve compelling results on this dataset. The extended dataset and code will be released soon.

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

↑