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

SuperPose:利用鲁棒跟踪与免掩码初始化改进的6D姿态估计

SuperPose: Improved 6D Pose Estimation with Robust Tracking and Mask-Free Initialization

Yu Deng, Jiahong Xue, Teng Cao, Yingxing Zhang, Lanxi Wen, Yiyang Chen

AI总结:

本文提出SuperPose,通过集成FoundationPose、SAM2和LightGlue,仅需CAD模型和一次点击即可实现免掩码初始化的鲁棒6D姿态估计与跟踪,有效解决跟踪丢失和对称物体问题。

摘要:

我们通过集成FoundationPose、SAM2和LightGlue,开发了一种用于工业应用中实时6D物体检测的鲁棒解决方案,无需重新训练。我们的方法解决了两个关键挑战:FoundationPose在第一帧中需要初始物体掩码的要求,以及对称物体在跟踪丢失和自动旋转方面的问题。该算法仅需要目标物体的CAD模型,用户在初始设置期间点击实时画面中物体的位置即可。设置完成后,算法会自动保存物体的参考图像,并在后续运行中使用LightGlue在物体与实时场景之间进行特征匹配,为检测提供初始提示。在YCB数据集以及漂白清洁剂和齿轮等工业部件上测试后,该算法展示了可靠的6D检测与跟踪能力。通过集成SAM2和FoundationPose,我们有效缓解了跟踪丢失等常见局限性,确保在遮挡或快速移动等挑战性条件下仍能进行连续且准确的跟踪。

英文摘要:

We developed a robust solution for real-time 6D object detection in industrial applications by integrating FoundationPose, SAM2, and LightGlue, eliminating the need for retraining. Our approach addresses two key challenges: the requirement for an initial object mask in the first frame in FoundationPose and issues with tracking loss and automatic rotation for symmetric objects. The algorithm requires only a CAD model of the target object, with the user clicking on its location in the live feed during the initial setup. Once set, the algorithm automatically saves a reference image of the object and, in subsequent runs, employs LightGlue for feature matching between the object and the real-time scene, providing an initial prompt for detection. Tested on the YCB dataset and industrial components such as bleach cleanser and gears, the algorithm demonstrated reliable 6D detection and tracking. By integrating SAM2 and FoundationPose, we effectively mitigated common limitations such as the problem of tracking loss, ensuring continuous and accurate tracking under challenging conditions like occlusion or rapid movement.

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

↑