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

TransPose:一种带深度细化的基于 Transformer 的 6D 物体位姿估计网络

TransPose: A Transformer-based 6D Object Pose Estimation Network with Depth Refinement

City, University of London(伦敦城市大学)

Mahmoud Abdulsalam, Nabil Aouf

AI总结:

本文提出仅用 RGB 输入的 Transformer 6D 位姿估计网络 TransPose,通过轻量深度估计、Transformer 检测头与深度细化模块提升精度,并在水果采摘场景中优于现有方法。

摘要:

随着机器人操作应用需求的增长,准确的基于视觉的 6D 位姿估计对于自主作业变得至关重要。此前已有基于卷积神经网络(CNNs)的位姿估计方法被提出。然而,对更优性能的追求仍在继续,尤其是为了实现精确的机器人操作;这一需求也延伸到农业机器人领域。本文提出 TransPose,这是一种改进的、基于 Transformer 的 6D 位姿估计方法,并配有深度细化模块。该架构仅接收 RGB 图像作为输入,不需要深度图像或热红外图像等额外补充模态。架构包含一个创新的轻量级深度估计网络,利用特征金字塔和上采样方法从 RGB 图像中估计深度。文中提出了一个带有额外预测头的、基于 Transformer 的检测网络,用于直接回归物体中心并预测目标的 6D 位姿。随后,结合预测得到的中心、6D 位姿和深度块,使用一个新颖的深度细化模块来提高所估计 6D 位姿的准确性。我们将所得结果与其他最先进方法进行了广泛比较,并针对水果采摘应用分析了结果。所取得的结果表明,本文提出的技术优于现有文献中的其他方法。

英文摘要:

As demand for robotics manipulation application increases, accurate vision-based 6D pose estimation becomes essential for autonomous operations. Convolutional Neural Networks (CNNs) based approaches for pose estimation have been previously introduced. However, the quest for better performance still persists especially for accurate robotics manipulation. This quest extends to the Agri-robotics domain. In this paper, we propose TransPose, an improved Transformer-based 6D pose estimation with a depth refinement module. The architecture takes in only an RGB image as input with no additional supplementing modalities such as depth or thermal images. The architecture encompasses an innovative lighter depth estimation network that estimates depth from an RGB image using feature pyramid with an up-sampling method. A transformer-based detection network with additional prediction heads is proposed to directly regress the object's centre and predict the 6D pose of the target. A novel depth refinement module is then used alongside the predicted centers, 6D poses and depth patches to refine the accuracy of the estimated 6D pose. We extensively compared our results with other state-of-the-art methods and analysed our results for fruit-picking applications. The results we achieved show that our proposed technique outperforms the other methods available in the literature.

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

↑