PCB 缺陷检测深度学习算法的精度改进
DOI:
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP182;TN911. 73

基金项目:

国家科技部重大科学仪器专项项目(2013YQ220749)资助


Accuracy improvement of deep learning algorithm for PCB defect detection
Author:
Affiliation:

Fund Project:

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

    本文以 YOLOv5 目标检测算法为基础算法,针对 PCB 缺陷检测进行了提高精度的改进。 首先通过实验选取了合适的 数据增强方法。 针对 PCB 缺陷尺寸小的问题,在原有的 3 个检测头基础上增加了 P2 检测头。 设计全新的 PANet 多特征融合 结构,实现高效的双向跨尺度连接和加权特征层融合。 针对 PCB 背景复杂的问题,引入了 CBAM 注意力模块以增强图像信息。 引入了 Transformer 模块来增强算法,以提高捕捉不同位置的 PCB 缺陷信息的能力。 最终通过这些改进,在检测速度 FPS 仅下 降 7. 2 的情况下,检测算法的 mAP 精度提高了 11. 3%。

    Abstract:

    In this paper, the YOLOv5 target detection algorithm was used as the base algorithm to improve accuracy for PCB defect detection. Firstly, an appropriate data augmentation method is selected through experiments. For the problem of small PCB defect size, the P2 detection head was added to the original three detection heads. A new PANet multi-feature fusion structure was designed to realize efficient two-way cross-scale connection and weighted feature layer fusion. For the problem of the complex PCB background, the CBAM attention module is introduced to enhance image information, the Transformer module is introduced to enhance the algorithm’s ability to capture PCB defect information at different locations. Finally, through these improvements, the mAP accuracy of the algorithm increased by 11. 3% while the FPS droped by only 7. 2.

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

杨永跃,夏远超. PCB 缺陷检测深度学习算法的精度改进[J].电子测量与仪器学报,2023,37(5):11-19

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