基于BCGS-YOLOv7 tiny的零件表面微小缺陷检测

Part surface micro defect detection based on BCGS-YOLOv7 tiny

  • 摘要: 为了解决零件表面众多形态复杂的微小缺陷检测效率低与检测精度不高的问题,选择以YOLOv7-tiny网络为基础架构进行改进,提出了BCGS-YOLOv7 tiny检测网络。特征提取阶段增加CBAM注意力机制,并用SPD-Conv下采样层代替跨步卷积和最大池化层,增强对小目标的特征提取能力;特征融合阶段将路径聚合特征金字塔(PAFPN)网络替换为BPANet网络结构,并引入gnCSP模块和SPD-Conv下采样层,改善小目标的特征融合能力。在重组的GC10-DET数据集上实验结果表明,BCGS -YOLOv7 tiny检测网络平均精度(mAP)达到 91.6%,比原YOLOv7-tiny网络提高了6.0%。同时对零件表面各类微小缺陷的检测精度均有大幅提升。

     

    Abstract: To address the issues of low efficiency and accuracy in detecting numerous complex micro defects on the surface of parts, an improved detection network called BCGS-YOLOv7 tiny is proposed, based on the YOLOv7-tiny architecture. In the feature extraction stage, the CBAM (Convolutional Block Attention Module) attention mechanism is incorporated, and the SPD-Conv downsampling layer is used instead of stride convolution and max-pooling layers to enhance the feature extraction capability for small objects. In the feature fusion stage, the Path Aggregation Feature Pyramid Network (PAFPN) is replaced by the BPANet network structure, and the gnCSP module and SPD-Conv downsampling layer are introduced to improve the feature fusion ability for small objects.Experimental results on the reconstructed GC10-DET dataset demonstrate that the BCGS-YOLOv7 tiny detection network achieves a mean Average Precision (mAP) of 91.6%, which is a 6.0% improvement over the original YOLOv7-tiny network. Moreover, the detection accuracy for various types of micro defects on part surfaces is significantly enhanced.

     

/

返回文章
返回