改进YOLOv8n的轻量化PCB缺陷检测算法
DOI:
CSTR:
作者:
作者单位:

重庆邮电大学通信与信息工程学院 重庆 400065

作者简介:

通讯作者:

中图分类号:

TN911.73;TN41

基金项目:

四川省科技计划项目(2023YFQ0019)资助


Improved lightweight PCB defect detection algorithm for YOLOv8n
Author:
Affiliation:

School of Engineering of Communication and Information,Chongqing University of Posts and Telecommunication, Chongqing 400065, China

Fund Project:

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

    针对现阶段PCB缺陷检测算法在模型参数量和检测精度无法同时兼顾的问题,本文提出改进YOLOv8n的轻量化PCB检测算法ST-YOLO。首先,采用轻量化主干网络StarNet替换YOLOv8n的主干网络,调整网络结构。删除大目标检测层,新增小目标检测层。其次,颈部网络中将C2f模块与Star Block和CA注意力机制结合设计出C2f-Star-CA模块,能够更好的融合局部和全局的上下文信息。最后,设计轻量化检测头,通过使用共享卷积减少模型的参数量。实验结果表明:与YOLOv8n相比,本文算法模型参数量减少了45.5%,计算量减少了56.8%,mAP%0.5提升了0.2%,为满足移动端部署的需要提供了新的可能性。

    Abstract:

    In view of the problem that the current PCB defect detection algorithm cannot simultaneously take into account the number of model parameters and detection accuracy, this paper proposes an improved lightweight PCB detection algorithm ST-YOLO based on YOLOv8n. First, the backbone network of YOLOv8n was replaced by the lightweight backbone network StarNet to adjust the network structure. Delete the large target detection layer and add the small target detection layer. Secondly, C2f module is combined with Star Block and CA attention mechanism to design C2f-Star-CA module, which can better integrate local and global context information. Finally, lightweight detection headers are designed to reduce the number of parameters in the model by using shared convolution. The experimental results show that compared with YOLOv8n, the number of model parameters is reduced by 45.5%, the calculation amount is reduced by 56.8%, and the mAP%0.5 is increased by 0.2%. It provides new possibilities to meet the needs of mobile deployment.

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

刘辉,刘旭,李校林,曾凡琪,王鹏江.改进YOLOv8n的轻量化PCB缺陷检测算法[J].电子测量技术,2025,48(13):111-119

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2025-08-04
  • 出版日期:
文章二维码

重要通知公告

①《电子测量技术》期刊收款账户变更公告