基于改进型YOLOv5的粉尘检测算法

    Dust detection algorithm based on improved YOLOv5

    • 摘要: 近年来,由于基于图像识别的粉尘检测方法不存在安装和检测范围局限性等问题,因此得到了充分重视和发展,但现有方法实时性和准确性仍需提升。为此,提出了一种基于改进YOLOv5算法的粉尘图像检测方法。首先,对现有YOLOv5算法主干网络以及Neck网络进行改进,将轻量化网络GhostNet替换原有主干网络,以降低网络参数,再输出3个特征层;然后,针对主干网络输出的3个特征层,施加注意力机制CA,增加网络精度;最后,设计消融实验和对比实验验证改进算法的有效性。结果表明:改进算法的平均检测精度mAP(mean Average Precision)能达到92.11%,检测速度达37帧/s。

       

      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 improved. 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 GhostNet, 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 that the mean Average Precision (mAP) of the improved algorithm can reach 92.11% and the detection speed reaches 37 frames persecond.

       

    /

    返回文章
    返回