基于关键点估计的抓取检测算法

摘要：

抓取是机器人在服务与工业领域中进行人机协调的重要能力,得到一个准确的抓取检测结果是机械臂能否完成抓取任务的关键.为了提高抓取检测的准确率以及实时性,提出了一种由CenterNet改进的基于关键点估计的抓取检测算法.在网络的特征提取层使用了特征融合方法融合不同的特征图,减少特征的丢失;增加了角度预测分支用来预测抓取角度;使用了改进的Focal Loss,减少由于正负样本不均衡导致的模型准确度降低.与基于锚框的抓取检测算法穷举目标潜在位置再进行回归的方式不同,基于关键点估计的抓取检测算法直接预测抓取关键点,并从关键点预测抓取框的尺寸、偏移量以及抓取角度.实验结果表明,与基于锚框的抓取检测相比,该方法更加高效、准确、简洁.在康奈尔数据集上,此模型达到了97.6％的准确率以及42 frame/s的检测速度.

关键词：关键点估计;抓取检测;目标检测;深度学习;康奈尔抓取数据集

所属期刊栏目：58

分类号：TP242(自动化技术及设备)

资助基金：国家重点研发计划2017YFB1301700

在线出版日期：2022-03-03（万方平台首次上网日期，不代表论文的发表时间）

页数：共8页

页码：267-274