机器人目标抓取区域实时检测方法

Real-Time Detection Method for Robot Target Grasping Area

针对目前机器人目标抓取区域检测方法无法兼顾检测准确率和实时性的问题,提出一种基于SE-RetinaGrasp神经网络模型的机器人目标抓取区域实时检测方法。该方法首先以一阶目标检测模型RetinaNet为基础提取抓取框位置及抓取角度;针对抓取检测任务采用SENet结构确定重要的特征通道;结合平衡特征金字塔设计思想,充分融合高低层的特征信息,以加强小抓取框的检测性能;在Cornell数据集上进行实验验证,结果表明该方法在取得更高检测准确率的同时,提高了抓取检测的效率,达到实时检测的要求。

Aiming at the problem that the current robot target grasping area detection method cannot take into account the detection accuracy and real-time performance, a real-time detection method of robot target grasping area based on SERetina Grasp neural network model is proposed. Firstly, based on the one-stage target detection model Retina Net, the position of the grasping rectangles and the grasping angle are extracted. Secondly, the SENet structure is used to determine the important feature channel in the grabbing detection task. Then, combined with the balanced feature pyramid design idea,the fusion is fully integrated. The feature information of the upper and lower layers is used to enhance the detection performance of the small grasping rectangles. Finally, the experimental verification is performed on the Cornell dataset. The results show that the method improves the detection accuracy while improving the detection accuracy, efficiency to meet real-time detection requirements.