YOLO-STOD: an industrial conveyor belt tear detection model based on Yolov5 algorithm

Abstract Real-time detection of conveyor belt tearing is of great significance to ensure mining in the coal industry. The longitudinal tear damage problem of conveyor belts has the characteristics of multi-scale, abundant small targets, and complex interference sources. Therefore, in order to improv...

Full description

Saved in:
Bibliographic Details
Main Authors: Wei Liu, Qing Tao, Nini Wang, Wendong Xiao, Cen Pan
Format: Article
Language:English
Published: Nature Portfolio 2025-01-01
Series:Scientific Reports
Subjects:
Online Access:https://doi.org/10.1038/s41598-024-83619-6
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1841544688088121344
author Wei Liu
Qing Tao
Nini Wang
Wendong Xiao
Cen Pan
author_facet Wei Liu
Qing Tao
Nini Wang
Wendong Xiao
Cen Pan
author_sort Wei Liu
collection DOAJ
description Abstract Real-time detection of conveyor belt tearing is of great significance to ensure mining in the coal industry. The longitudinal tear damage problem of conveyor belts has the characteristics of multi-scale, abundant small targets, and complex interference sources. Therefore, in order to improve the performance of small-size tear damage detection algorithms under complex interference, a visual detection method YOLO-STOD based on deep learning was proposed. Firstly, a multi-case conveyor belt tear dataset is developed for complex interference and small-size detection. Second, the detection method YOLO-STOD is designed, which utilizes the BotNet attention mechanism to extract multi-dimensional tearing features, enhancing the model’s feature extraction ability for small targets and enables the model to converge quickly under the conditions of few samples. Secondly, Shape_IOU is utilized to calculate the training loss, and the shape regression loss of the bounding box itself is considered to enhance the robustness of the model. The experimental results fully proved the effectiveness of the YOLO-STOD detection method, which constantly surpasses the competing methods and achieves 91.2%, 91.9%, and 190.966 detection accuracy and detection speed in terms of recall, Map value, and FPS, respectively, which is able to satisfy the needs of industrial real-time detection and is expected to be used in the real-time detection of conveyor belt tearing in the industrial field.
format Article
id doaj-art-a6a31e51c0714fab9624aacf37bcaf5a
institution Kabale University
issn 2045-2322
language English
publishDate 2025-01-01
publisher Nature Portfolio
record_format Article
series Scientific Reports
spelling doaj-art-a6a31e51c0714fab9624aacf37bcaf5a2025-01-12T12:22:05ZengNature PortfolioScientific Reports2045-23222025-01-0115112110.1038/s41598-024-83619-6YOLO-STOD: an industrial conveyor belt tear detection model based on Yolov5 algorithmWei Liu0Qing Tao1Nini Wang2Wendong Xiao3Cen Pan4School of Mechanical Engineering, Xinjiang UniversitySchool of Mechanical Engineering, Xinjiang UniversityCollege of Electrical Engineering, Xinjiang UniversitySchool of Mechanical Engineering, Xinjiang UniversitySchool of Mechanical Engineering, Xinjiang UniversityAbstract Real-time detection of conveyor belt tearing is of great significance to ensure mining in the coal industry. The longitudinal tear damage problem of conveyor belts has the characteristics of multi-scale, abundant small targets, and complex interference sources. Therefore, in order to improve the performance of small-size tear damage detection algorithms under complex interference, a visual detection method YOLO-STOD based on deep learning was proposed. Firstly, a multi-case conveyor belt tear dataset is developed for complex interference and small-size detection. Second, the detection method YOLO-STOD is designed, which utilizes the BotNet attention mechanism to extract multi-dimensional tearing features, enhancing the model’s feature extraction ability for small targets and enables the model to converge quickly under the conditions of few samples. Secondly, Shape_IOU is utilized to calculate the training loss, and the shape regression loss of the bounding box itself is considered to enhance the robustness of the model. The experimental results fully proved the effectiveness of the YOLO-STOD detection method, which constantly surpasses the competing methods and achieves 91.2%, 91.9%, and 190.966 detection accuracy and detection speed in terms of recall, Map value, and FPS, respectively, which is able to satisfy the needs of industrial real-time detection and is expected to be used in the real-time detection of conveyor belt tearing in the industrial field.https://doi.org/10.1038/s41598-024-83619-6BELT tearingYolo algorithmReal-time detectionDeep learningSmall object detection
spellingShingle Wei Liu
Qing Tao
Nini Wang
Wendong Xiao
Cen Pan
YOLO-STOD: an industrial conveyor belt tear detection model based on Yolov5 algorithm
Scientific Reports
BELT tearing
Yolo algorithm
Real-time detection
Deep learning
Small object detection
title YOLO-STOD: an industrial conveyor belt tear detection model based on Yolov5 algorithm
title_full YOLO-STOD: an industrial conveyor belt tear detection model based on Yolov5 algorithm
title_fullStr YOLO-STOD: an industrial conveyor belt tear detection model based on Yolov5 algorithm
title_full_unstemmed YOLO-STOD: an industrial conveyor belt tear detection model based on Yolov5 algorithm
title_short YOLO-STOD: an industrial conveyor belt tear detection model based on Yolov5 algorithm
title_sort yolo stod an industrial conveyor belt tear detection model based on yolov5 algorithm
topic BELT tearing
Yolo algorithm
Real-time detection
Deep learning
Small object detection
url https://doi.org/10.1038/s41598-024-83619-6
work_keys_str_mv AT weiliu yolostodanindustrialconveyorbeltteardetectionmodelbasedonyolov5algorithm
AT qingtao yolostodanindustrialconveyorbeltteardetectionmodelbasedonyolov5algorithm
AT niniwang yolostodanindustrialconveyorbeltteardetectionmodelbasedonyolov5algorithm
AT wendongxiao yolostodanindustrialconveyorbeltteardetectionmodelbasedonyolov5algorithm
AT cenpan yolostodanindustrialconveyorbeltteardetectionmodelbasedonyolov5algorithm