论文标题
咬合感知的自我监督单眼6D物体姿势估计
Occlusion-Aware Self-Supervised Monocular 6D Object Pose Estimation
论文作者
论文摘要
6D对象姿势估计是计算机视觉中的一个基本而又具有挑战性的问题。卷积神经网络(CNN)最近已被证明能够预测可靠的6D姿势估计,即使在单眼环境下也是如此。尽管如此,CNN被认为是非常数据驱动的,并且获取足够的注释通常非常耗时且劳动量很大。为了克服这一局限性,我们通过自我监督的学习提出了一种新颖的单眼6D姿势估计方法,从而消除了对真实注释的需求。在培训了我们提出的网络通过合成的RGB数据进行了充分监督之后,我们利用了嘈杂的学生培训和可区分渲染的当前趋势,以进一步自我避免这些无监督的真实RGB(-D)样本,以寻求视觉和几何学上最佳的一致性。此外,使用可见的和阿莫达尔面具的信息,我们的自我审视变得非常强大,在诸如遮挡之类的挑战性场景上变得非常强大。广泛的评估表明,我们提出的自我实施者优于依靠合成数据或采用来自域自适应领域的精心技术的所有其他方法。值得注意的是,我们的自我监督方法在其合成训练的基线方面持续改善,并且通常几乎缩小了完全监督的差距。代码和模型可在https://github.com/thu-da-6d-pose-group/self6dpp.git上公开获取。
6D object pose estimation is a fundamental yet challenging problem in computer vision. Convolutional Neural Networks (CNNs) have recently proven to be capable of predicting reliable 6D pose estimates even under monocular settings. Nonetheless, CNNs are identified as being extremely data-driven, and acquiring adequate annotations is oftentimes very time-consuming and labor intensive. To overcome this limitation, we propose a novel monocular 6D pose estimation approach by means of self-supervised learning, removing the need for real annotations. After training our proposed network fully supervised with synthetic RGB data, we leverage current trends in noisy student training and differentiable rendering to further self-supervise the model on these unsupervised real RGB(-D) samples, seeking for a visually and geometrically optimal alignment. Moreover, employing both visible and amodal mask information, our self-supervision becomes very robust towards challenging scenarios such as occlusion. Extensive evaluations demonstrate that our proposed self-supervision outperforms all other methods relying on synthetic data or employing elaborate techniques from the domain adaptation realm. Noteworthy, our self-supervised approach consistently improves over its synthetically trained baseline and often almost closes the gap towards its fully supervised counterpart. The code and models are publicly available at https://github.com/THU-DA-6D-Pose-Group/self6dpp.git.