{"status":"ok","message-type":"work","message-version":"1.0.0","message":{"indexed":{"date-parts":[[2026,4,22]],"date-time":"2026-04-22T23:11:26Z","timestamp":1776899486385,"version":"3.51.2"},"reference-count":23,"publisher":"MDPI AG","issue":"11","license":[{"start":{"date-parts":[[2024,5,28]],"date-time":"2024-05-28T00:00:00Z","timestamp":1716854400000},"content-version":"vor","delay-in-days":0,"URL":"https:\/\/creativecommons.org\/licenses\/by\/4.0\/"}],"funder":[{"name":"National Natural Science Foundation of China","award":["61960206010"],"award-info":[{"award-number":["61960206010"]}]}],"content-domain":{"domain":[],"crossmark-restriction":false},"short-container-title":["Sensors"],"abstract":"<jats:p>Train wheels are crucial components for ensuring the safety of trains. The accurate and fast identification of wheel tread defects is necessary for the timely maintenance of wheels, which is essential for achieving the premise of conditional repair. Image-based detection methods are commonly used for detecting tread defects, but they still have issues with the misdetection of water stains and the leaking of small defects. In this paper, we address the challenges posed by the detection of wheel tread defects by proposing improvements to the YOLOv8 model. Firstly, the impact of water stains on tread defect detection is avoided by optimising the structure of the detection layer. Secondly, an improved SPPCSPC module is introduced to enhance the detection of small targets. Finally, the SIoU loss function is used to accelerate the convergence speed of the network, which ensures defect recognition accuracy with high operational efficiency. Validation was performed on the constructed tread defect dataset. The results demonstrate that the enhanced YOLOv8 model in this paper outperforms the original network and significantly improves the tread defect detection indexes. The average precision, accuracy, and recall reached 96.95%, 96.30%, and 95.31%.<\/jats:p>","DOI":"10.3390\/s24113477","type":"journal-article","created":{"date-parts":[[2024,5,28]],"date-time":"2024-05-28T07:38:39Z","timestamp":1716881919000},"page":"3477","update-policy":"https:\/\/doi.org\/10.3390\/mdpi_crossmark_policy","source":"Crossref","is-referenced-by-count":13,"title":["Improved YOLOv8-Based Target Precision Detection Algorithm for Train Wheel Tread Defects"],"prefix":"10.3390","volume":"24","author":[{"given":"Yu","family":"Wen","sequence":"first","affiliation":[{"name":"School of Physical Science and Technology, Southwest Jiaotong University, Chengdu 610031, China"}],"role":[{"role":"author","vocabulary":"crossref"}]},{"given":"Xiaorong","family":"Gao","sequence":"additional","affiliation":[{"name":"School of Physical Science and Technology, Southwest Jiaotong University, Chengdu 610031, China"}],"role":[{"role":"author","vocabulary":"crossref"}]},{"given":"Lin","family":"Luo","sequence":"additional","affiliation":[{"name":"School of Physical Science and Technology, Southwest Jiaotong University, Chengdu 610031, China"}],"role":[{"role":"author","vocabulary":"crossref"}]},{"ORCID":"https:\/\/orcid.org\/0000-0003-2724-7807","authenticated-orcid":false,"given":"Jinlong","family":"Li","sequence":"additional","affiliation":[{"name":"School of Physical Science and Technology, Southwest Jiaotong University, Chengdu 610031, China"}],"role":[{"role":"author","vocabulary":"crossref"}]}],"member":"1968","published-online":{"date-parts":[[2024,5,28]]},"reference":[{"key":"ref_1","doi-asserted-by":"crossref","first-page":"103679","DOI":"10.1016\/j.trc.2022.103679","article-title":"A Literature Review of Artificial Intelligence Applications in Railway Systems","volume":"140","author":"Tang","year":"2022","journal-title":"Transp. Res. Part C Emerg. Technol."},{"key":"ref_2","doi-asserted-by":"crossref","unstructured":"Girshick, R., Donahue, J., Darrell, T., and Malik, J. (2014, January 23\u201328). Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Columbus, OH, USA.","DOI":"10.1109\/CVPR.2014.81"},{"key":"ref_3","doi-asserted-by":"crossref","first-page":"1137","DOI":"10.1109\/TPAMI.2016.2577031","article-title":"Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks","volume":"39","author":"Ren","year":"2017","journal-title":"IEEE Trans. Pattern Anal. Mach. Intell."},{"key":"ref_4","doi-asserted-by":"crossref","first-page":"103551","DOI":"10.1016\/j.compind.2021.103551","article-title":"Improved Faster R-CNN for Fabric Defect Detection Based on Gabor Filter with Genetic Algorithm Optimization","volume":"134","author":"Chen","year":"2022","journal-title":"Comput. Ind."},{"key":"ref_5","doi-asserted-by":"crossref","unstructured":"Xu, X., Zhao, M., Shi, P., Ren, R., He, X., Wei, X., and Yang, H. (2022). Crack Detection and Comparison Study Based on Faster R-CNN and Mask R-CNN. Sensors, 22.","DOI":"10.3390\/s22031215"},{"key":"ref_6","unstructured":"Redmon, J., Divvala, S., Girshick, R., and Farhadi, A. (July, January 26). You Only Look Once: Unified, Real-Time Object Detection. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA."},{"key":"ref_7","doi-asserted-by":"crossref","first-page":"12559","DOI":"10.1038\/s41598-022-16302-3","article-title":"End-to-End Deep Learning Framework for Printed Circuit Board Manufacturing Defect Classification","volume":"12","author":"Bhattacharya","year":"2022","journal-title":"Sci. Rep."},{"key":"ref_8","doi-asserted-by":"crossref","first-page":"102722","DOI":"10.1016\/j.asej.2024.102722","article-title":"Hypertuned-YOLO for Interpretable Distribution Power Grid Fault Location Based on EigenCAM","volume":"15","author":"Stefenon","year":"2024","journal-title":"Ain Shams Eng. J."},{"key":"ref_9","doi-asserted-by":"crossref","unstructured":"Ding, F., Zhuang, Z., Liu, Y., Jiang, D., Yan, X., and Wang, Z. (2020). Detecting Defects on Solid Wood Panels Based on an Improved SSD Algorithm. Sensors, 20.","DOI":"10.3390\/s20185315"},{"key":"ref_10","doi-asserted-by":"crossref","first-page":"36","DOI":"10.1007\/s10044-024-01252-5","article-title":"CABF-YOLO: A Precise and Efficient Deep Learning Method for Defect Detection on Strip Steel Surface","volume":"27","author":"Zhou","year":"2024","journal-title":"Pattern Anal. Appl."},{"key":"ref_11","doi-asserted-by":"crossref","unstructured":"Hussain, M. (2023). YOLO-v1 to YOLO-v8, the Rise of YOLO and Its Complementary Nature toward Digital Manufacturing and Industrial Defect Detection. Machines, 11.","DOI":"10.3390\/machines11070677"},{"key":"ref_12","doi-asserted-by":"crossref","unstructured":"Fan, Y., Mao, S., Li, M., Wu, Z., and Kang, J. (2024). CM-YOLOv8: Lightweight YOLO for Coal Mine Fully Mechanized Mining Face. Sensors, 24.","DOI":"10.20944\/preprints202401.1814.v1"},{"key":"ref_13","doi-asserted-by":"crossref","unstructured":"Sampurno, R.M., Liu, Z., Abeyrathna, R.M.R.D., and Ahamed, T. (2024). Intrarow Uncut Weed Detection Using You-Only-Look-Once Instance Segmentation for Orchard Plantations. Sensors, 24.","DOI":"10.3390\/s24030893"},{"key":"ref_14","unstructured":"Selvaraju, R.R., Das, A., Vedantam, R., Cogswell, M., Parikh, D., and Batra, D. (2017). Grad-CAM: Why Did You Say That?. arXiv."},{"key":"ref_15","doi-asserted-by":"crossref","first-page":"1904","DOI":"10.1109\/TPAMI.2015.2389824","article-title":"Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition","volume":"37","author":"He","year":"2015","journal-title":"IEEE Trans. Pattern Anal. Mach. Intell."},{"key":"ref_16","doi-asserted-by":"crossref","unstructured":"Wang, C.-Y., Mark Liao, H.-Y., Wu, Y.-H., Chen, P.-Y., Hsieh, J.-W., and Yeh, I.-H. (2020, January 14\u201319). CSPNet: A New Backbone That Can Enhance Learning Capability of CNN. Proceedings of the 2020 IEEE\/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), Seattle, WA, USA.","DOI":"10.1109\/CVPRW50498.2020.00203"},{"key":"ref_17","unstructured":"Yang, L., Zhang, R.-Y., Li, L., and Xie, X. (2021, January 18\u201324). SimAM: A Simple, Parameter-Free Attention Module for Convolutional Neural Networks. Proceedings of the 38th International Conference on Machine Learning, Virtual."},{"key":"ref_18","doi-asserted-by":"crossref","unstructured":"Rezatofighi, H., Tsoi, N., Gwak, J., Sadeghian, A., Reid, I., and Savarese, S. (2019, January 15\u201320). Generalized Intersection Over Union: A Metric and a Loss for Bounding Box Regression. Proceedings of the IEEE\/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA.","DOI":"10.1109\/CVPR.2019.00075"},{"key":"ref_19","first-page":"12993","article-title":"Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression","volume":"34","author":"Zheng","year":"2020","journal-title":"Proc. AAAI Conf. Artif. Intell."},{"key":"ref_20","doi-asserted-by":"crossref","first-page":"146","DOI":"10.1016\/j.neucom.2022.07.042","article-title":"Focal and Efficient IOU Loss for Accurate Bounding Box Regression","volume":"506","author":"Zhang","year":"2022","journal-title":"Neurocomputing"},{"key":"ref_21","unstructured":"Zhora, G. (2022). SIoU Loss: More Powerful Learning for Bounding Box Regression. arXiv."},{"key":"ref_22","doi-asserted-by":"crossref","first-page":"21","DOI":"10.1007\/978-3-319-46448-0_2","article-title":"SSD: Single Shot MultiBox Detector","volume":"Volume 9905","author":"Leibe","year":"2016","journal-title":"Computer Vision\u2014ECCV 2016"},{"key":"ref_23","unstructured":"Redmon, J., and Farhadi, A. (2018). Yolov3: An Incremental Improvement. arXiv."}],"container-title":["Sensors"],"original-title":[],"language":"en","link":[{"URL":"https:\/\/www.mdpi.com\/1424-8220\/24\/11\/3477\/pdf","content-type":"unspecified","content-version":"vor","intended-application":"similarity-checking"}],"deposited":{"date-parts":[[2025,10,10]],"date-time":"2025-10-10T14:49:32Z","timestamp":1760107772000},"score":1,"resource":{"primary":{"URL":"https:\/\/www.mdpi.com\/1424-8220\/24\/11\/3477"}},"subtitle":[],"short-title":[],"issued":{"date-parts":[[2024,5,28]]},"references-count":23,"journal-issue":{"issue":"11","published-online":{"date-parts":[[2024,6]]}},"alternative-id":["s24113477"],"URL":"https:\/\/doi.org\/10.3390\/s24113477","relation":{},"ISSN":["1424-8220"],"issn-type":[{"value":"1424-8220","type":"electronic"}],"subject":[],"published":{"date-parts":[[2024,5,28]]}}}