高级检索

基于深度学习与ROS分布式的机器人智能抓取系统

Robot Intelligent Grasping System Based on Deep Learning and ROS Distribution

  • 摘要: 提出了一种结合深度学习与ROS的智能抓取系统。该系统基于YOLOv5算法模型,引入MobileNetV3作为网络骨干,有效提升了模型的轻量化程度;将颈部网络的Conv替换为组洗牌卷积(Group Shuffle Convolution, GSConv),以降低模型复杂度并增强注意力机制的效果;将颈部网络中的C3模块替换为VoVGSCSP,进一步降低网络复杂度。在最后一个检测头前添加并行卷积注意力模块(Parallel Convolutional Block Attention Module, PCBAM),增强了模型对目标物体的敏感度。与YOLOv5s基线模型相比,改进模型召回率提升3.1%,参数量下降49.6%,运算量下降65.8%。实验结果表明,该系统检测实时性高,在杂乱无序场景下也能准确识别并抓取物体。

     

    Abstract: In this paper, an intelligent grasping system combining deep learning and ROS is proposed. The system is based on the YOLOv5 algorithm model, and MobileNetV3 is introduced as the network backbone, which effectively improves the lightweight degree of the model. The Conv of the neck network was replaced by the group shuffle convolution(GSConv) to reduce the complexity of the model and enhance the effect of the attention mechanism. The C3 module in the neck network is replaced with VoVGSCSP to further reduce network complexity. The parallel convolutional block attention module(PCBAM) is added before the last detection head to enhance the model′s sensitivity to the target object. Compared with the YOLOv5s baseline model, the recall rate of the improved model is increased by 3.1%, the number of parameters is reduced by 49.6%, and the amount of computation is reduced by 65.8%. Experimental results show that the system has high real-time detection performance, and can accurately identify and grasp objects in chaotic scenes.

     

/

返回文章
返回