改进YOLOv5的气缸盖锻件缺陷视觉检测方法研究

Research on visual detection method of cylinder head forging defects based on improved YOLOv5

  • 摘要: 汽车气缸盖锻件在生产过程中容易出现各种裂纹,影响产品质量。针对当前气缸盖锻件检测缺陷精度和效率低的问题,提出了一种基于YOLOv5网络改进的算法模型YOLOv5-MG。首先,搭建图像采集平台,制作气缸盖锻件缺陷样本数据集。然后,将YOLOv5 Head架构SPP-YOLO替换为Decoupled Head结构,提高模型的准确率和效率;使用双向特征金字塔(bidirectional feature pyramid network, BiFPN)替换YOLOv5网络原来的路径聚合网络(path aggregation network, PANet),增强图像浅层特征信息与深层特征信息的融合;引入完备交并(complete intersection over union, CIoU)的计算方法提升算法检测的定位精度。实验表明,改进的算法在测试集上的均值平均精度达到了99.81%,F1的分值为0.99,浮点运算数为17.2 B。相较于其他深度学习模型,该方法有效地提高了气缸盖锻件的缺陷检测效率和精度,能够满足缺陷检测的要求。

     

    Abstract: Various types of cracks are prone to be produced in the production process of automotive cylinder head forgings, which affect the product quality. An improved algorithm model YOLOv5-MG based on the YOLOv5 network is proposed to address the current problem of low accuracy and efficiency in detecting defects in cylinder head forgings. Firstly, a sample dataset of cylinder head forgings defects is produced by building an image acquisition platform. Then, the accuracy and efficiency of the model are improved by replacing the YOLOv5 Head architecture of the SPP-YOLO network with the Decoupled Head structure; the fusion of shallow feature information with deeper feature information is enhanced by replacing the original path aggregation network (PANet) of the YOLOv5 network with the bidirectional feature pyramid network (BFPN); and the localization accuracy of the algorithm is improved by introducing the computation method of complete intersection over union (CIoU). 99.81% mean accuracy, 0.99 F1 score and 17.2 B floating point operations (FLOPs) are achieved by the improved algorithm on the test set.

     

/

返回文章
返回