Baomin Wang, Hewei Ding, Fei Teng, Zhirong Wang, Hongqin Liu
{"title":"基于改进型 YOLOv5 的钢丝绳芯输送带损坏物体检测技术","authors":"Baomin Wang, Hewei Ding, Fei Teng, Zhirong Wang, Hongqin Liu","doi":"10.1142/s0219467825500573","DOIUrl":null,"url":null,"abstract":"In response to the challenges in detecting damage features in X-ray images of steel wire rope-cores in conveyor belts, such as complex damage shapes, small sizes, low detection precision, and poor generalization ability, an improved YOLOv5 algorithm was proposed. The aim of the model is to accurately and efficiently identify and locate damage in the X-ray images of steel wire rope-cores in conveyor belts. First, the Adaptive Histogram Equalization (AHE) method is used to preprocess the images, reducing the interference of harsh mining environments and improving the quality of the dataset. Second, to better retain image details and enhance the detection ability of damage features, transpose convolutional upsampling is adopted, and the C3 module in the backbone network is replaced by C2f to ensure lightweight network models, meanwhile, it obtains richer gradient flow information and optimizing the loss function. Finally, the improved algorithm is compared with four classical detection algorithms using the damage feature dataset of steel wire rope-core conveyor belts. The experimental result shows that the proposed algorithm achieves an average detection precision of 91.8% and a detection speed of 40 frames per second (FPS) for images collected in harsh mining environments. The designed detection model provides a reference for the automatic recognition and detection of damage to steel wire rope-core conveyor belts.","PeriodicalId":44688,"journal":{"name":"International Journal of Image and Graphics","volume":null,"pages":null},"PeriodicalIF":0.8000,"publicationDate":"2024-03-25","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":"{\"title\":\"Damage Object Detection of Steel Wire Rope-Core Conveyor Belts Based on the Improved YOLOv5\",\"authors\":\"Baomin Wang, Hewei Ding, Fei Teng, Zhirong Wang, Hongqin Liu\",\"doi\":\"10.1142/s0219467825500573\",\"DOIUrl\":null,\"url\":null,\"abstract\":\"In response to the challenges in detecting damage features in X-ray images of steel wire rope-cores in conveyor belts, such as complex damage shapes, small sizes, low detection precision, and poor generalization ability, an improved YOLOv5 algorithm was proposed. The aim of the model is to accurately and efficiently identify and locate damage in the X-ray images of steel wire rope-cores in conveyor belts. First, the Adaptive Histogram Equalization (AHE) method is used to preprocess the images, reducing the interference of harsh mining environments and improving the quality of the dataset. Second, to better retain image details and enhance the detection ability of damage features, transpose convolutional upsampling is adopted, and the C3 module in the backbone network is replaced by C2f to ensure lightweight network models, meanwhile, it obtains richer gradient flow information and optimizing the loss function. Finally, the improved algorithm is compared with four classical detection algorithms using the damage feature dataset of steel wire rope-core conveyor belts. The experimental result shows that the proposed algorithm achieves an average detection precision of 91.8% and a detection speed of 40 frames per second (FPS) for images collected in harsh mining environments. The designed detection model provides a reference for the automatic recognition and detection of damage to steel wire rope-core conveyor belts.\",\"PeriodicalId\":44688,\"journal\":{\"name\":\"International Journal of Image and Graphics\",\"volume\":null,\"pages\":null},\"PeriodicalIF\":0.8000,\"publicationDate\":\"2024-03-25\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"\",\"citationCount\":\"0\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"International Journal of Image and Graphics\",\"FirstCategoryId\":\"1085\",\"ListUrlMain\":\"https://doi.org/10.1142/s0219467825500573\",\"RegionNum\":0,\"RegionCategory\":null,\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"Q4\",\"JCRName\":\"COMPUTER SCIENCE, SOFTWARE ENGINEERING\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"International Journal of Image and Graphics","FirstCategoryId":"1085","ListUrlMain":"https://doi.org/10.1142/s0219467825500573","RegionNum":0,"RegionCategory":null,"ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q4","JCRName":"COMPUTER SCIENCE, SOFTWARE ENGINEERING","Score":null,"Total":0}
引用次数: 0
摘要
针对传送带钢丝绳芯 X 射线图像中损伤形状复杂、尺寸小、检测精度低、泛化能力差等损伤特征检测难题,提出了改进的 YOLOv5 算法。该模型旨在准确、高效地识别和定位传送带中钢丝绳芯 X 射线图像中的损伤。首先,采用自适应直方图均衡化(AHE)方法对图像进行预处理,减少恶劣采矿环境的干扰,提高数据集的质量。其次,为了更好地保留图像细节,提高损伤特征的检测能力,采用了转置卷积上采样,并将骨干网络中的 C3 模块替换为 C2f,保证了网络模型的轻量化,同时获得了更丰富的梯度流信息,优化了损失函数。最后,利用钢丝绳芯输送带的损伤特征数据集,将改进算法与四种经典检测算法进行了比较。实验结果表明,对于在恶劣采矿环境中采集的图像,所提出的算法实现了 91.8% 的平均检测精度和每秒 40 帧(FPS)的检测速度。所设计的检测模型为自动识别和检测钢丝绳芯输送带的损坏提供了参考。
Damage Object Detection of Steel Wire Rope-Core Conveyor Belts Based on the Improved YOLOv5
In response to the challenges in detecting damage features in X-ray images of steel wire rope-cores in conveyor belts, such as complex damage shapes, small sizes, low detection precision, and poor generalization ability, an improved YOLOv5 algorithm was proposed. The aim of the model is to accurately and efficiently identify and locate damage in the X-ray images of steel wire rope-cores in conveyor belts. First, the Adaptive Histogram Equalization (AHE) method is used to preprocess the images, reducing the interference of harsh mining environments and improving the quality of the dataset. Second, to better retain image details and enhance the detection ability of damage features, transpose convolutional upsampling is adopted, and the C3 module in the backbone network is replaced by C2f to ensure lightweight network models, meanwhile, it obtains richer gradient flow information and optimizing the loss function. Finally, the improved algorithm is compared with four classical detection algorithms using the damage feature dataset of steel wire rope-core conveyor belts. The experimental result shows that the proposed algorithm achieves an average detection precision of 91.8% and a detection speed of 40 frames per second (FPS) for images collected in harsh mining environments. The designed detection model provides a reference for the automatic recognition and detection of damage to steel wire rope-core conveyor belts.