-
公开(公告)号:CN118691574A
公开(公告)日:2024-09-24
申请号:CN202410797049.8
申请日:2024-06-20
Inventor: 刘丹
IPC: G06T7/00 , G06V10/774 , G06V10/82 , G06V10/80 , G06V10/766 , G06V10/764
Abstract: 本发明公开一种PCB小目标缺陷检测方法,获取数据集,并按预设比例随机划分为训练集和验证集;构建改进的YOLOv7‑tiny网络模型,改进的YOLOv7‑tiny网络模型包括主干网络、颈部网络和头部网络,主干网络用于提取不同语义特征并进行特征级联,颈部网络采用GLCM模块来捕获主干网络的多特征图的全局和局部信息;头部网络使用单一尺度的预测头网络进行类别分类和边界框的回归;通过训练集和验证集训练改进的YOLOv7‑tiny网络模型,输出PCB小目标缺陷检测网络;将待检测PCB表面图片输入至PCB小目标缺陷检测网络,获得PCB缺陷小目标的定位信息。本发明能够提高预测精度且满足实时速度。