Dust detection algorithm based on improved YOLOv5
CHEN Qinghua
ZHANG Junwei
CHENG Yingsong
ZHANG Xu
CHENG Jianhua
Abstract：In recent years,dust detection methods based on image recognition have received full attention and development because they do not have installation and detection range limitations,but the real-time and accuracy of existing methods still need to be im-proved.In view of this,we propose a dust image detection method based on the improved YOLOv5 algorithm.Firstly,the existing YOLOv5 algorithm backbone network and Neck network were improved,and the original backbone network was replaced by Ghost-Net,a lightweight network,to reduce network parameters,and then three feature layers were output.Then,for the three feature layers of the backbone network output,the attention mechanism CA is applied to increase the network accuracy.Finally,ablation experiments and comparative experiments were designed to verify the effectiveness of the improved algorithm.The experimental results show th-at the mean Average Precision(mAP)of the improved algorithm can reach 92.11%and the detection speed reaches 37 frames persecond.
Keywords：dust image detectionimproved YOLOv5 algorithmconfidence coefficientLightweight networkCA attention mech-anism
Publication Date：2025-06-20
Online Publishing Date：2025-08-15(First online date of this platform, not the publication date of the document)
Pages：10( 79-88 )