改进 YOLOv5s 的 PCB 元器件检测技术

来源：机械与电子杂志2024年第10期北京时间：

作者：张晋钊;刘新妹;殷俊龄1

单位：1. 中北大学电子测试技术国家重点实验室,山西 太原 030051 ;

摘要：针对 PCB 元器件检测中硬件运算需求高、检测速度慢等问题,提出一种基于 YOLOv5s 改进的轻量化 PCB 元器件目标检测算法 YOLO-SCGS 。该算法以 ShuffleNetv2 替换主干网络,在 Neck 层引入 CA 注意力机制,使用轻量级卷积方法代替标准卷积,加入 SPD-Conv 重构 CNN 体系结构,代替卷积步长和卷积层,提高检测精度。实验结果表明,所提算法相较原 YOLOv5s 检测速度提升 16.61% ,浮点计算需求减少 85.44% ,模型权重缩小 84.83% ,有更低的运算需求和更优的检测速度,满足轻量化本地部署的要求。

关键词：YOLOv5s ;印制电路板;目标检测;轻量化

基金资助：山西省重点研发项目( 201903D121058 )