深度学习的接触网小目标缺陷识别研究
DOI:
作者:
作者单位:

兰州交通大学自动化与电气工程学院兰州730070

作者简介:

通讯作者:

中图分类号:

U225.4;TP391.4

基金项目:

甘肃省科技计划资助项目(20JR10RA216)资助


Identification of the catenary small target defects in deep learning
Author:
Affiliation:

School of Automation and Electrical Engineering, Lanzhou Jiaotong University, Lanzhou 730070, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    吊弦线夹螺栓是铁路接触网供电线路的重要器件,其状态会影响电力机车受流质量,于是对SSD算法进行改进:首先引入一种轻量级神经网络MobileNetV3用于前端特征提取,降低模型复杂度,以提高检测速度;其次采用CA注意力机制替换反向残差结构线性瓶颈层的SE模块,使位置信息沿空间两个方向聚合,调整后的特征层能够捕获全局远程特征信息;最后设计了特征融合模块以重构特征层,优化小目标检测层以提高对小目标的识别效果。还用CycleGAN等方法扩充训练样本,解决数据集不足的问题。实验结果表明,改进算法的模型复杂度下降,mAP@0.5和FPS分别达到95.5%和81 fps,该研究有助于接触网检测仪器向小型移动嵌入式设备转变。

    Abstract:

    The dropper clamp bolt is an important component of railway power supply line, which can affect the flow quality of electric locomotive. Therefore, this paper improves the SSD algorithm: Firstly, a lightweight neural network MobileNetV3 is introduced for front-end feature extraction to reduce the model complexity and improve the detection speed; secondly, CA attention mechanism to replace the SE module of the linear bottleneck layer with inverted residuals structure, aggregate the position information in the two directions of space, and the adjusted feature layer can capture the global remote feature information. Finally, the feature fusion module for reconstructing the feature layer is designed to adjust the small target detection layer to improve the detection effect of small targets. This paper also expands the training sample with CycleGAN to solve the problem of insufficient data set. The experimental results show that the model complexity of the improved algorithm decreased, and mAP @ 0.5 and FPS reached 95.5% and 81 fps, respectively. This study helps the transformation of catenary detection instruments to small mobile embedded devices.

    参考文献
    相似文献
    引证文献
引用本文

顾桂梅,王小亮.深度学习的接触网小目标缺陷识别研究[J].电子测量与仪器学报,2024,38(4):151-160

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-07-02
  • 出版日期: