Open Access iconOpen Access

ARTICLE

crossmark

YOLO-VSI: An Improved YOLOv8 Model for Detecting Railway Turnouts Defects in Complex Environments

Chenghai Yu, Zhilong Lu*

School of Computer Science and Technology, Zhejiang Sci-Tech University, Hangzhou, 310018, China

* Corresponding Author: Zhilong Lu. Email: email

Computers, Materials & Continua 2024, 81(2), 3261-3280. https://doi.org/10.32604/cmc.2024.056413

Abstract

Railway turnouts often develop defects such as chipping, cracks, and wear during use. If not detected and addressed promptly, these defects can pose significant risks to train operation safety and passenger security. Despite advances in defect detection technologies, research specifically targeting railway turnout defects remains limited. To address this gap, we collected images from railway inspectors and constructed a dataset of railway turnout defects in complex environments. To enhance detection accuracy, we propose an improved YOLOv8 model named YOLO-VSS-SOUP-Inner-CIoU (YOLO-VSI). The model employs a state-space model (SSM) to enhance the C2f module in the YOLOv8 backbone, proposed the C2f-VSS module to better capture long-range dependencies and contextual features, thus improving feature extraction in complex environments. In the network’s neck layer, we integrate SPDConv and Omni-Kernel Network (OKM) modules to improve the original PAFPN (Path Aggregation Feature Pyramid Network) structure, and proposed the Small Object Upgrade Pyramid (SOUP) structure to enhance small object detection capabilities. Additionally, the Inner-CIoU loss function with a scale factor is applied to further enhance the model’s detection capabilities. Compared to the baseline model, YOLO-VSI demonstrates a 3.5% improvement in average precision on our railway turnout dataset, showcasing increased accuracy and robustness. Experiments on the public NEU-DET dataset reveal a 2.3% increase in average precision over the baseline, indicating that YOLO-VSI has good generalization capabilities.

Keywords


Cite This Article

APA Style
Yu, C., Lu, Z. (2024). YOLO-VSI: an improved yolov8 model for detecting railway turnouts defects in complex environments. Computers, Materials & Continua, 81(2), 3261-3280. https://doi.org/10.32604/cmc.2024.056413
Vancouver Style
Yu C, Lu Z. YOLO-VSI: an improved yolov8 model for detecting railway turnouts defects in complex environments. Comput Mater Contin. 2024;81(2):3261-3280 https://doi.org/10.32604/cmc.2024.056413
IEEE Style
C. Yu and Z. Lu, “YOLO-VSI: An Improved YOLOv8 Model for Detecting Railway Turnouts Defects in Complex Environments,” Comput. Mater. Contin., vol. 81, no. 2, pp. 3261-3280, 2024. https://doi.org/10.32604/cmc.2024.056413



cc Copyright © 2024 The Author(s). Published by Tech Science Press.
This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 341

    View

  • 143

    Download

  • 0

    Like

Share Link