首页 | 本学科首页   官方微博 | 高级检索  
     

改进 YOLOv8 的轨道扣件状态检测方法
引用本文:范华琦,杨 柳.改进 YOLOv8 的轨道扣件状态检测方法[J].国外电子测量技术,2024,43(4):19-29.
作者姓名:范华琦  杨 柳
作者单位:1.西南交通大学信息科学与技术学院
基金项目:轨道交通工程信息化国家重点实验室(中铁一院)开放课题(SKLKZ22-02)、宜宾市双城市校协议专项科研经费科 技项目(SWJTU2021020005)、 朔黄铁路公司科研项目(SHYP-22-01) 资助
摘    要:针对现有轨道扣件状态检测算法对小目标物体与复杂形状物体的检测能力差而导致检测结果异常,以及小目标层特 征冗余等问题,提出了一种改进 YOLOv8 的轨道扣件状态检测方法。在 YOLOv8 网络中增加可变形空间金字塔扩张卷积模 块,以提高模型对小目标物体以及形变复杂物体的检测精度。同时增加小目标空间重构单元以减少小目标特征冗余,促进小 目标特征的学习。根据采集到的轨道扣件数据集进行模型的训练和测试,并与多组轨道扣件状态检测算法进行对比,实验结 果表明,相较于对比算法,所提算法精确度平均提升3.20%,召回率平均提升3.34%,平均精度平均提升3.96%。实验证明 所提算法能够有效进行轨道轨道扣件状态检测,并且具有较强的泛化能力,可以部署于复杂交通场景。

关 键 词:图像处理  YOLOv8    轨道扣件  小目标检测

Improved YOLOv8 track fastener state detection method
Fan Huaqi,Yang Liu.Improved YOLOv8 track fastener state detection method[J].Foreign Electronic Measurement Technology,2024,43(4):19-29.
Authors:Fan Huaqi  Yang Liu
Affiliation:1. (School of Information Science and Technology,Southwest Jiaotong University,
Abstract:Aiming at the problems of poor detection ability of existing track fastenings state detection algorithms on small target objects and complex shapes,which lead to abnormal detection results,as well as feature redundancy of small target layer,an improved YOLOv8 track fastenings state detection method was proposed.A deformable space pyramid expansion convolutional module is added in YOLOv8 network to improve the detection accuracy of small target objects and deformable complex objects.At the same time,the reconstruction unit of small target space is added to reduce the redundancy of small target features and promote the learning of small target features.The model is trained and tested according to the collected data set of track fasteners,and compared with multiple groups of track fasteners state detection algorithms.The experimental results show that compared with the comparison algorithm,the accuracy of the proposed algorithm is increased by 3.20%on average,the recall rate is increased by 3.34%on average,and the average accuracy is increased by 3.96%on average.The experiments proves that the proposed algorithm can detect the state of rail fastenings effectively,and has strong generalization ability,and can be deployed in complex traffic scenarios.
Keywords:image    processing  YOLOv8  track    fastener  small      target    detection
点击此处可从《国外电子测量技术》浏览原始摘要信息
点击此处可从《国外电子测量技术》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号