提速6D物体姿态估计算法,保持高精度。
FAST GDRNPP: Improving the Speed of State-of-the-Art 6D Object Pose Estimation
- 用小模型骨干+剪枝+知识蒸馏优化GDRNPP
- 推理速度显著提升,精度仍达顶尖水平
- 适合工业场景中的实时机器人操作
6D物体姿态估计需确定物体在场景中的三维平移和旋转,对工业质检、抓取、机器人操作等应用至关重要。当前深度学习模型常面临精度与延迟的权衡。本文针对先进模型GDRNPP,通过采用更小更快的骨干网络、剪枝冗余参数以及知识蒸馏,将大型高性能模型的知识迁移至更轻量的学生模型,有效压缩模型规模并加速推理。实验表明,新配置在保持与现有最优模型相当精度的同时,显著提升推理效率,为实际工业场景中6D姿态估计的应用提供了更高效率的解决方案。
原文摘要 · Abstract (English)
6D object pose estimation involves determining the three-dimensional translation and rotation of an object within a scene and relative to a chosen coordinate system. This problem is of particular interest for many practical applications in industrial tasks such as quality control, bin picking, and robotic manipulation, where both speed and accuracy are critical for real-world deployment. Current models, both classical and deep-learning-based, often struggle with the trade-off between accuracy and latency. Our research focuses on enhancing the speed of a prominent state-of-the-art deep learning model, GDRNPP, while keeping its high accuracy. We employ several techniques to reduce the model size and improve inference time. These techniques include using smaller and quicker backbones, pruning unnecessary parameters, and distillation to transfer knowledge from a large, high-performing model to a smaller, more efficient student model. Our findings demonstrate that the proposed configuration maintains accuracy comparable to the state-of-the-art while significantly improving inference time. This advancement could lead to more efficient and practical applications in various industrial scenarios, thereby enhancing the overall applicability of 6D Object Pose Estimation models in real-world settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。