改进PSO-YOLOv5的金属表面缺陷检测方法  

Research on Metal Surface Defect Detection by Improved PSO-YOLOv5

作　　者：蔡聪艺 Cai Congyi(Zhangzhou Institute of Technology,Zhangzhou 363000,China)

机构地区：[1]漳州职业技术学院智能制造学院,福建漳州363000

出　　处：《廊坊师范学院学报（自然科学版）》2022年第4期37-41,共5页Journal of Langfang Normal University(Natural Science Edition)

基　　金：福建省中青年教育科研项目(JZ180803);漳州职业技术学院科研项目(ZZY2021B050)。

摘　　要：针对已有检测方法在目标尺寸较小时适应度差及特征不清晰导致漏检等问题,提出改进的PSO-YOLOv5金属缺陷检测方法。基于YOLOv5网络结合PSO算法优化网络的颈部部分,同时加入了自适应锚框计算,为边界框提供更准确的移动方向和位置信息。最后使用反向传播神经网络对图像进行训练得到最终的深度卷积神经网络。与传统检测方法相比,改进后的PSO-YOLOv5算法检测速率为31.6frame/s,平均检测精度为78.56%,较YOLOv5提高8.32个百分点,检测精度优于Faster RCNN等算法,表明改进的PSO-YOLOv5算法能够有效提高小目标情况下的缺陷检测精度。Aiming at the problems that existing detection methods have poor adaptability and unclear features when the target size is small,this paper proposes an improved PSO-YOLOv5 metal defect detection method.This method is based on the YOLOv5 network and combined with the PSO algorithm to optimize the neck part of the network.At the same time,the adaptive anchor box calculation is added to provide more accurate moving direction and more accurate position information for the bounding box.Finally,the back-propagation neural network is used to train the image to obtain the final deep convolutional neural network.Compared with the traditional detection method,the improved PSO-YOLOv5 algorithm has a detection performance of 31.6 frame/s and an average detection accuracy of 78.56%,which is 8.32 percentage points higher than that of YOLOv5.The detection accuracy is better than that of Faster RCNN and other algorithms.The results show that the improved PSO-YOLOv5 can effectively improve the defect detection accuracy in the case of small targets.

关 键 词：目标检测 金属表面缺陷 YOLOv5 PSO 距离交并比(DIoU) 

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