摘要
arXiv:2409.12720v1 公告类型: 交叉 摘要: 6D物体姿态估计涉及确定物体在场景中的三维平移和旋转,并相对于选定的坐标系进行定位。这一问题在许多工业任务中尤为重要,如质量控制、料箱拣选和机器人操作,这些应用中速度和精度对于实际部署至关重要。当前的模型,无论是经典的还是基于深度学习的,通常在精度和延迟之间难以平衡。我们的研究专注于提升一种先进的深度学习模型GDRNPP的速度,同时保持其高精度。我们采用了多种技术来减小模型尺寸并提高推理时间,包括使用更小、更快的骨干网络,修剪不必要的参数,以及通过蒸馏将知识从大型高性能模型转移到更小、更高效的学生模型。我们的研究结果表明,所提出的配置在保持与最先进技术相当的精度的同时,显著提高了推理时间。这一进展可能会在各种工业场景中实现更高效和实用的应用,从而增强6D物体姿态估计模型在实际环境中的整体适用性。