基于改进YOLOv5的电子粉涂覆不均检测  

Improved YOLOv5-based for detection of uneven coating of electronic powders

作　　者：戴思璇 何青[1] 唐琼霜 洪巍 Dai Sixuan;He Qing;Tang Qiongshuang;Hong Wei(School of Electrical&Information Engineering,Changsha University of Science&Technology,Changsha 410114,China)

机构地区：[1]长沙理工大学电气与信息工程学院,长沙410114

出　　处：《电子测量技术》2023年第14期155-161,共7页Electronic Measurement Technology

基　　金：国家自然科学基金(62103063)项目资助

摘　　要：放电管生产过程中,电极表面电子粉涂覆是否均匀是影响放电管产品质量的关键,目前主要依赖于人工目检,针对人工检测效率低、精度差等问题,提出一种基于改进YOLOv5的电子粉涂覆不均检测算法。首先,采集电极表面电子粉涂覆的图像制作数据集,并进行数据增强;其次,引入STDC模块优化主干特征提取网络,提高对难以辨认的金属电极表面缺陷不均的检测精度,并生成两个特征图以适应数据集;最后使用K-means++聚类优化自适应锚框计算。实验结果表明:改进YOLOv5算法对电子粉涂覆不均检测的mAP@50达到99.22%,与原YOLOv5网络相比提升了6.84%,大大提升了检测精度,相较于人工检测其效率更高。In the process of gas discharge tube production,the uniformity of electronic powder coating on the electrode surface is the key to the quality of gas discharge tube products,it is mainly detected by human eyes now.Aiming at the problems of low efficiency and poor accuracy of manual detection,an uneven electronic powder coating detection algorithm based on improved YOLOv5is proposed.Firstly,the collected images of electron powder coating on the electrode surface are made into data sets,and data enhancement was performed.Secondly,the STDC module is used to optimize the backbone feature extraction network,to improve the detection accuracy of uneven surface defects of hardto-recognize metal electrodes,and two feature layers are generated for adapting to the dataset size.Finally,Kmeans++clustering is used to optimize the computation of adaptive anchor boxes.The experimental results show that the mAP@50of the improved YOLOv5algorithm proposed reaches 99.22%,which is 6.84%higher than that of the original YOLOv5network,greatly improving the detection accuracy,and is more efficient than manual detection.

关 键 词：放电管 缺陷检测 YOLOv5 STDC模块 K-means++聚类 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术]