河北工业科技2024,Vol.41Issue(5) :321-329.DOI:10.7535/hbgykj.2024yx05001

基于改进YOLOv5的火车连接钩舌识别方法

Recognition method of train connection hook tongue based on improved YOLOv5

王冬伟 武彬 宋刚 高硕 韩昊宇 丁佳毅 董帆 万书亭
河北工业科技2024,Vol.41Issue(5) :321-329.DOI:10.7535/hbgykj.2024yx05001

基于改进YOLOv5的火车连接钩舌识别方法

Recognition method of train connection hook tongue based on improved YOLOv5

王冬伟 1武彬 1宋刚 1高硕 1韩昊宇 1丁佳毅 2董帆 2万书亭2
扫码查看

作者信息

  • 1. 安徽华电宿州发电有限公司,安徽宿州 234000
  • 2. 河北省电力机械装备健康维护与失效预防重点实验室,河北保定 071003
  • 折叠

摘要

为了准确识别不同类型钩舌,确保自动复钩机器人能够根据火车连接钩舌状态实时调整机械臂的位姿,提出了 一种基于改进YOLOv5的火车连接钩舌识别方法.首先,将YOLOv5主干网络中原有的C3模块替换为梯度流丰富的C2F模块(cross feature module),YOLOv5颈部网络中原有的C3模块替换为基于FasterNet模块构建的轻量化C3_FasterNet模块,并将CoordConv模块嵌入到YOLOv5的主干网络末端.其次,基于现场实测的火车连接钩舌图像进行了识别测试.结果表明:改进的YOLOv5算法在降低模型参数量的同时,可以有效提升对钩舌目标的检测精度,火车钩舌识别精度达到了 98.7%,相较于原始算法,模型参数量减少了 10.8%.研究结果为复钩机器人在执行钩舌复位和车厢连接操作方面提供了一种有效的解决方案.

Abstract

In order to accurately identify different types of train connection hook tongues and ensure that the automatic re-hook robot adjusts the pose of the robot arm in real time according to the status of the hook tongues,a train connection hook tongue recognition method based on an improved YOLOv5 model was proposed.First,the original C3 module in the YOLOv5 backbone network was replaced with the C2F module(cross feature module)of rich gradient flow,and the original C3 module in the YOLOv5 neck network was replaced with the lightweight C3_FasterNet module based on the FasterNet block,and the CoordConv module was embedded at the end of the YOLOv5 backbone network.Second,the recognition test was carried out based on the spot measured image of the train connection hook tongue.The results show that the improved YOLOv5 algorithm can effectively improve the detection accuracy of the hook tongue target while reducing the numbers of model parameters.The identification accuracy of the hook tongue reaches 98.7%,and the numbers of model parameters are reduced by 10.8%compared with the original algorithm,which can provide an effective solution for the re-hook robot in the operation of hook tongue resetting and carriage connection.

关键词

模式识别/图像处理/复钩机器人/火车钩舌/目标识别/YOLOv5

Key words

pattern recognition/image processing/re-hook robot/train connection hook tongue/target recognition/YOLOv5

引用本文复制引用

基金项目

国家自然科学基金(52275109)

河北省自然科学基金(E2022502007)

出版年

2024
河北工业科技
河北科技大学

河北工业科技

CSTPCD
影响因子:0.694
ISSN:1008-1534
参考文献量15
段落导航相关论文