基于改进YOLOv5的轨道异物入侵检测算法研究

Research on orbital foreign object intrusion detection algorithm based on improved YOLOv5

  • 摘要: 针对铁路场景图像不清晰,以及存在小目标和被遮挡目标未被检测出等问题,提出一种基于改进YOLOv5的异物目标检测算法. 首先,计算图像中的直线特征极大值调整自适应参数,利用边界点权重区分出轨道位置;然后,采用CLAHE(Contrast Limited Adaptive Histogram Equalization)算法增强画面的对比度,降低环境对检测结果的影响;最后,在原有YOLOv5算法的基础上引入卷积块注意力模型,提高特征提取能力,改善遮挡目标和小目标的漏检问题,用CIOU损失函数替代GIOU损失函数作为边界框回归损失函数,加快模型的收敛速度并提高边框定位精度. 实验结果表明,模型平均检测精度达到了0.941,检测速度也达到了39 帧/s,可以快速且准确地检测到铁路上存在的异物,满足实时目标检测的要求.

     

    Abstract: The track foreign object detection algorithm based on improved YOLOv5 is proposed for problems such as unclear images in railroad scenes and the existence of small targets and obscured targets that are not detected. Firstly, the adaptive parameters are adjusted according to the maximum value of the line characteristics of the scene, and the rail area is divided by the weight of the boundary points. Then, for the problem that the image is not clear, the CLAHE(Contrast-limited Adaptive Histogram Equalization)is used to improve the contrast of the picture and reduce the external influence. Finally, on the basis of the original YOLOv5 algorithm, the CBAM module is introduced to improve the feature extraction ability and the leakage detection of occluded targets and small targets. The CIOU loss function is used to replace the GIOU loss function as the bounding box regression loss function to accelerate the convergence speed of the model and improve the accuracy of border localization. The experimental results show that the model achieves an average detection accuracy of mean average precision of 0.941 and detection speed of 39 frames per second, which can quickly and accurately detect the foreign objects present on the railroad and meet the requirements of real-time target detection.

     

/

返回文章
返回