高级检索

基于改进YOLOv8的弓网接触点检测算法研究

Research on Contact Point Detection Algorithm of Bow Net Based on Improved YOLOv8

  • 摘要: 针对高速列车在运行中弓网接触点检测困难的问题,提出了一种基于改进YOLOv8的YOLOv8-HTPCP检测算法。该算法在主干网络引入Swim-transformerV2,提高模型的训练稳定性和准确性,对全局有更好的信息提取能力;使用了GSConv来减轻模型的复杂度并保持准确性;将轻量型的Ghost模块与YOLOv8算法相结合,可以大幅度降低网络参数量。在弓网接触点的数据集上进行消融实验和对比实验,实验结果表明,YOLOv8-HTPCP模型的平均精度均值mAP0.5值为98.5%,mAP0.5~0.95值为97.8%,召回率为96.7%,参数量2659823,与原模型YOLOv8相比,mAP0.5值、mAP0.5~0.95值、召回率分别提高了1.6、3.9、1.8个百分点,且参数量降低了14.7个百分点,为弓网几何参数参数检测提供了技术参考。

     

    Abstract: The proposed YOLOv8-HTPCP detection algorithm, based on an enhanced version of YOLOv8, addresses the challenge of detecting contact points in the arch mesh of high-speed trains. This algorithm incorporates Swim-transformerV2 into the backbone network to enhance model training stability and accuracy, enabling improved information extraction from the entire scene. GSConv is utilized to reduce complexity while maintaining model accuracy. Furthermore, integrating a lightweight Ghost module with the YOLOv8 algorithm significantly reduces network parameters. Ablation and comparison experiments were conducted using an arch net contact point dataset. The results demonstrate that the average accuracy of the YOLOv8-HTPCP model reached 98.5% for mAP0.5 value, 97.8% for mAP0.5 ~ 0.95 value, and 96.7% for reference number 2659823. Compared to the original YOLOv8 model, there was a respective increase of 1.6%, 3.9%, and 1.8% in mAP0.5 value, mAP0.5 ~ 0.95 value ,and recall rate; meanwhile, there was a decrease of 14.7% in parameter count.This study provides valuable technical insights for detecting geometric parameters in bow meshes.

     

/

返回文章
返回