基于改进YOLOv5s模型的印刷电路板缺陷检测方法

    公开(公告)号:CN116152169A

    公开(公告)日:2023-05-23

    申请号:CN202211607983.6

    申请日:2022-12-14

    Inventor: 陈炫睿 周应华

    Abstract: 本发明涉及一种基于改进YOLOv5s模型的印刷电路板缺陷检测方法,属于印刷电路板缺陷检测领域。该方法包括:获取选择印刷电路板缺陷数据集;改进YOLOv5s网络结构的Backbone部分,在中小目标特征提取层嵌入CA注意力机制;改进YOLOv5s网络结构的Neck部分,在特征融合层嵌入CBAM卷积注意力机制;使用Alpha‑CIOU损失函数代替原始YOLOv5s模型的CIOU损失函数;S3:对改进后的YOLOv5s网络模型进行训练;S4:随机选择缺陷数据集中的缺陷图片,使用训练好的网络模型进行检测,输出待检测缺陷目标的位置信息,类别信息以及置信度数值。本发明提高了检测印刷电路板上各种缺陷的精度。

Patent Agency Ranking