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

抓取是机器人在服务与工业领域中进行人机协调的重要能力,得到一个准确的抓取检测结果是机械臂能否完成抓取任务的关键.为了提高抓取检测的准确率以及实时性,提出了一种由CenterNet改进的基于关键点估计的抓取检测算法.在网络的特征提取层使用了特征融合方法融合不同的特征图,减少特征的丢失;增加了角度预测分支用来预测抓取角度;...

Grasping Detection Based on Key Point Estimation

Grasping is an important capability of human-machine coordination for robots in both service and industrial scenes. Obtaining an accurate grasp detection result is the key for the manipulator to complete the grasp task. In order to improve the accuracy and real-time performance of grasp detection, a grasp detection algorithm based on key point estimation which is improved from CenterNet is proposed. Firstly, in the feature extraction layer of the network, the feature fusion method is used to fuse different feature graphs for reducing feature loss. Secondly, the angle prediction branch is added to predict the grasp angle. Finally, improved Focal Loss is used to reduce the reduction of accuracy caused by the imbalance of positive and negative samples. Different from the anchor-based grasp detection algorithm, which enumerates the potential location of the object and then the object is measured by regression method. The key point estimation method directly predicts the grasp key points, then based on the key points, predicts the size, offset and angle of the target. The experimental results show that compared with the anchor-based grasp detection, the proposed method is more efficient, accurate and simple. The model achieves 97.6% accuracy and 42 frame/s detection speed on the Cornell grasp dataset.