|
TrackDef-YOLO:一种改进的YOLOv11模型用于铁轨表面缺陷检测
|
Abstract:
随着铁路运输的快速发展,铁路轨道的安全性与可靠性成为重要的研究课题。传统的人工检测方法效率低且易受人为因素影响。近年来,深度学习,尤其是YOLO系列模型在目标检测中的应用取得了显著进展。为提高铁轨表面缺陷检测的精度和实时性,本文提出了一种改进的YOLOv11模型(TrackDef-YOLO)。通过引入多尺度特征融合、自动锚框调整和加权损失函数等创新技术,TrackDef-YOLO显著提升了对微小缺陷和复杂背景的检测能力。实验结果表明,TrackDef-YOLO相比YOLOv8和YOLOv11n模型在准确率、召回率、F1-Score和误报率等方面均有显著提升。该研究为铁路轨道智能检测提供了高效、精准的解决方案,并为基于深度学习的铁路维护系统的优化奠定了基础。
With the rapid development of railway transportation, the safety and reliability of railway tracks have emerged as critical research topics. Traditional manual inspection methods are inefficient and susceptible to human factors. In recent years, deep learning, especially YOLO series models, has achieved remarkable progress in object detection. To enhance the accuracy and real-time performance of railway track surface defect detection, this paper proposes an improved YOLOv11 model, named TrackDef-YOLO. By incorporating innovative techniques such as multi-scale feature fusion, automatic anchor box adjustment, and a weighted loss function, TrackDef-YOLO significantly improves the detection capability for minute defects and complex backgrounds. Experimental results demonstrate that TrackDef-YOLO outperforms YOLOv8 and YOLOv11n models in terms of accuracy, recall, F1-Score, and false alarm rate. This study provides an efficient and precise solution for intelligent railway track inspection and lays the foundation for optimizing railway maintenance systems based on deep learning.
[1] | Wu, Y., Qin, Y., Wang, Z. and Jia, L. (2018) A UAV-Based Visual Inspection Method for Rail Surface Defects. Applied Sciences, 8, Article 1028. https://doi.org/10.3390/app8071028 |
[2] | Gan, J., Li, Q., Wang, J. and Yu, H. (2017) A Hierarchical Extractor-Based Visual Rail Surface Inspection System. IEEE Sensors Journal, 17, 7935-7944. https://doi.org/10.1109/jsen.2017.2761858 |
[3] | Chen, G., Wang, X., Ni, W. and Wei, L. (2023) A Steel Rail Surface Defect Detection and Recognition Method Based on Mask R-Transformer. 2023 IEEE 6th International Conference on Pattern Recognition and Artificial Intelligence (PRAI), Haikou, China, 18-20 August 2023. https://doi.org/10.1109/prai59366.2023.10332076 |
[4] | Wang, Y., Zhang, K., Wang, L. and Wu, L. (2024) An Improved YOLOv8 Algorithm for Rail Surface Defect Detection. IEEE Access, 12, 44984-44997. https://doi.org/10.1109/access.2024.3380009 |
[5] | Zhao, J., Yeung, A.W., Ali, M., Lai, S. and Ng, V.T. (2024) CBAM-Swint-BL: Small Rail Surface Defect Detection Method Based on Swin Transformer with Block Level CBAM Enhancement. IEEE Access, 12, 181997-182009. https://doi.org/10.1109/access.2024.3509986 |
[6] | Han, H., Tian, L., Li, M., Cui, X., Shang, C. and Hou, S. (2023). Design of Rail Surface Defect Detection System Based on LabVIEW Machine Vision. 2023 IEEE 6th Information Technology, Networking, Electronic and Automation Control Conference (ITNEC), Chongqing, China, 24-26 February 2023. https://doi.org/10.1109/itnec56291.2023.10082124 |
[7] | Zhang, C., Xu, D., Zhang, L. and Deng, W. (2023) Rail Surface Defect Detection Based on Image Enhancement and Improved YOLOX. Electronics, 12, Article 2672. https://doi.org/10.3390/electronics12122672 |
[8] | Du, J., Zhang, R., Gao, R., Nan, L. and Bao, Y. (2024) RSDNet: A New Multiscale Rail Surface Defect Detection Model. Sensors, 24, Article 3579. https://doi.org/10.3390/s24113579 |
[9] | Fang, Z., Li, L., Peng, L., Zheng, S., Zhong, Q. and Zhu, T. (2024) YOLOv8n-RSDD: A High-Performance Low-Complexity Rail Surface Defect Detection Network. IEEE Access, 12, 196249-196265. https://doi.org/10.1109/access.2024.3466559 |
[10] | Bochkovskiy, A., Wang, C.Y, Liao, H.Y.M., et al. (2020) YOLOv4: Optimal Speed and Accuracy of Object Detection. arXiv preprint arXiv: 2004.10934. |
[11] | Lin, T.-Y., Dollar, P., Girshick, R., et al. (2017) Feature Pyramid Networks for Object Detection. 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, 21-26 July 2017, 936-944. |
[12] | Lin, T.Y., Goyal, P., Girshick, R., He, K.M. and Dollar, P. (2018) Focal Loss for Dense Object Detection. IEEE Transactions on Pattern Analysis and Machine Intelligence, 42, 318-327. |
[13] | Rezatofighi, H., Tsoiet, N., Gwak, GY., et al. (2019) Generalized Intersection over Union: A Metric and A Loss for Bounding Box Regression. 2019 IEEE Transactions on Pattern Analysis and Machine Intelligence, Long Beach, 15-20 June 2019, 658-666. https://doi.org/10.1109/CVPR.2019.00075 |