计算机工程与应用 ›› 2020, Vol. 56 ›› Issue (19): 224-230.DOI: 10.3778/j.issn.1002-8331.1907-0214
卢智亮,林伟,曾碧,刘瑞雪
LU Zhiliang, LIN Wei, ZENG Bi, LIU Ruixue
摘要:
针对目前机器人目标抓取区域检测方法无法兼顾检测准确率和实时性的问题,提出一种基于SE-RetinaGrasp神经网络模型的机器人目标抓取区域实时检测方法。该方法首先以一阶目标检测模型RetinaNet为基础提取抓取框位置及抓取角度;针对抓取检测任务采用SENet结构确定重要的特征通道;结合平衡特征金字塔设计思想,充分融合高低层的特征信息,以加强小抓取框的检测性能;在Cornell数据集上进行实验验证,结果表明该方法在取得更高检测准确率的同时,提高了抓取检测的效率,达到实时检测的要求。