基于改进Faster RCNN的轮对踏面缺陷检测
DOI:
CSTR:
作者:
作者单位:

1.西南交通大学机械工程学院 成都 610031; 2.西南交通大学牵引动力国家重点实验室 成都 610031

作者简介:

通讯作者:

中图分类号:

U216.3

基金项目:

博士后科学基金 (2020M682506)、成都科技局重点研发项目(2019-YF05-01823-SN)资助


Defect detection of wheelset tread based on improved Faster RCNN
Author:
Affiliation:

1.School of Mechanical Engineering, Southwest Jiaotong University, Chengdu 610031, China; 2.State Key Laboratory of Traction Power, Southwest Jiaotong University, Chengdu 610031,China

Fund Project:

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

    针对目前传统图像处理算法对踏面缺陷检测存在效率不高、对环境鲁棒性不足等问题,本文提出基于改进Faster RCNN的踏面缺陷检测方法。改进的网络首先使用Resnet50作为特征提取网络,并在特征金字塔层(FPN)特征融合输出部分加入自注意力机制,加强了检测网络对小缺陷的检测能力,最后使用K-means++聚类算法对踏面缺陷数据集锚框进行聚类,并通过聚类结果定制出更适合轮对踏面缺陷的锚框。实验结果表明,改进后的Faster RCNN网络对轮对踏面缺陷检测的平均检测速度为68 ms,平均精度(mAP)达到了97.3%,对小目标缺陷的检测精度(mAPsmall)达到了39.3%。

    Abstract:

    To address the problems of inefficiency and lack of robustness to the environment in the current traditional image processing algorithms for tread surface defect detection, this paper proposes an improved tread surface defect detection method based on the Faster RCNN. The improved network first uses Resnet50 as the feature extraction network, and adds a self-attention mechanism to the feature fusion output part of the Feature Pyramid Network to enhance the detection ability of the detection network for small defects, and finally uses the K-means++ clustering algorithm to cluster the anchor frames of the tread defect dataset, and uses the clustering results to customize anchor frames that are more suitable for wheel-to-tread defects.The experimental results show that the improved Faster RCNN network has an average detection speed of 68 ms, an average accuracy (mAP) of 97.3% and an accuracy of 39.3% for the detection of small target defects.

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

刘应桃,郭世伟,付孟新,张青松.基于改进Faster RCNN的轮对踏面缺陷检测[J].电子测量技术,2023,46(12):34-41

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