基于YOLO算法的高速铁路客运车站钢结构雨棚螺栓缺失检测系统  

Missing bolt detection system for steel structure canopy of high-speed railway passenger station based on YOLO algorithm

作　　者：王域辰 冯海龙 刘伯奇 WANG Yuchen;FENG Hailong;LIU Boqi(Railway Engineering Research Institute,China Academy of Railway Sciences Corporation Limited,Beijing 100081,China)

机构地区：[1]中国铁道科学研究院集团有限公司铁道建筑研究所,北京100081

出　　处：《铁路计算机应用》2022年第6期1-5,共5页Railway Computer Application

基　　金：中国国家铁路集团有限公司科技研究开发计划(J2020G003)。

摘　　要：当前,我国高速铁路客运车站钢结构雨棚螺栓缺失检测过于依赖人工目测,其危险系数大、成本高、效率低且误检率高。为解决该问题,提出一种基于YOLO(You Only Look Once)算法的螺栓缺失检测系统。该系统采用YOLOv4卷积神经网络,对现场采集的钢结构雨棚和接触网螺栓进行标注,通过K-means聚类算法,确定锚框数目和尺寸;利用CutMix和Mosaic等数据增强操作,增加训练数据的多样性,避免出现训练过拟合。试验结果表明,该系统类别识别准确率可达85%以上,识别效果较好,满足检测实时性要求。At present,the missing detection of steel structure canopy bolts in high-speed railway passenger stations in China relies too much on manual visual inspection,which has great risk coefficient,high cost,low efficiency and high false detection rate.In order to solve this problem,this paper proposed a bolt missing detection system based on YOLO algorithm.The system used YOLO v4 convolution neural network to mark the steel structure canopy and catenary bolts collected on site,determined the number and size of anchor frames by K-means clustering algorithm,and used cutmix,mosaic and other data to enhance the operation,increase the diversity of training data,and avoid training over fitting.The trial results show that the category recognition accuracy of the system can reach more than 85%,and the recognition effect is good,which meets the requirements of real-time detection.

关 键 词：YOLOv4算法 钢结构雨棚 螺栓检测 深度学习 数据增强 

分 类 号：U291.63[交通运输工程—交通运输规划与管理] TP39[交通运输工程—道路与铁道工程]