基于改进YOLOv8n的 PCB 缺陷检测算法
DOI:
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP391.4

基金项目:

湖北省重点研发计划(2021BAA193) 项目资助


PCB defect detection algorithm based on improved YOLOv8n
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对现有的印刷电路板(PCB) 缺陷检测方法计算量大、小目标缺陷易漏检、检测速度较慢等问题,提出 YOLOv8n- 4SCDP缺陷检测算法。首先,在YOLOv8n 颈部网络增加上采样,融合 Backbone 中浅层语义信息,同时增加微小目标检测层 降低PCB小目标缺陷漏检率;其次,在Backbone 中融入坐标注意力(CA) 机制,强化特征语义和位置信息,提高了模型特征融 合能力;另外,设计密集连接机构,提高模型的缺陷特征利用率,采用PConv对模型进行压缩,既保证了模型的准确性,又大大 减小了模型的尺寸;最后,针对难易样本不平衡的问题,采用线性区间映射法重新定义回归损失函数(Focaler-SIoU),提高模 型收敛速度和回归精度。实验结果表明,YOLOv8n-4SCDP 算法的整体缺陷的平均精度均值(mAP) 达到95.8%,检测帧率达 到了65 fps。有效改善YOLOv8n 对于PCB小目标缺陷漏检率高、检测精度低等问题。

    Abstract:

    In response to the issues of large computational load,easy omission of small target defects,and slow detection speed in existing PCB defect detection methods,this paper proposes the YOLOv8n-4SCDP defect detection algorithm. Firstly,upsampling is added to the neck network of YOLOv8n,integrating shallow semantic information in the Backbone,and a small target detection head is added to reduce the omission rate of small target defects in PCBs. Secondly,the CA attentionmechanism is integrated into the Backbone to enhance the semantic and positional information of features,thereby improving the feature fusion capability of the model.Thirdly,a dense connection mechanism was designed to enhance the utilization of defect features in the model.Additionally,PConv was employed to compress the model,ensuring both accuracy and significantly reducing the model's size.Finally,to address the issue of imbalanced difficult and easy samples,we employ a linear interval mapping method to redefine the Focaler-SIoU regression loss function.This approach enhances both model convergence speed and regression accuracy.The experimental results indicate that the YOLOv8n-4SCDP algorithm achieves an accuracy of 95.8%and a frame rate of 65 fps.This effectively addresses YOLOv8n's issues related to high defect omission rates and low detection accuracy for small PCB targets.

    参考文献
    相似文献
    引证文献
引用本文

姜 源,付 波,权 轶,李 吴.基于改进YOLOv8n的 PCB 缺陷检测算法[J].国外电子测量技术,2024,43(6):22-32

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-07-09
  • 出版日期:
文章二维码