一种基于改进SSD算法的工件表面缺陷检测方法

    公开(公告)号:CN116071331A

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

    申请号:CN202310108176.8

    申请日:2023-02-13

    申请人: 淮阴工学院

    摘要: 本发明公开了一种基于改进SSD算法的工件表面缺陷检测方法,采用数据集增强算法对预先获取的工件表面缺陷数据集进行预处理;搭建基于改进SSD算法的工件表面缺陷检测模型,包括基础网络层、特征映射层、特征预测层和非极大值抑制模块;基础网络层是将轻量级神经网络MobileNetv2替换VGG‑16作为SSD的基础网络;在特征映射层增加两个并行卷积模块,引入深度可分离卷积、空洞卷积、跳跃连接和像素加法操作,并且进行小、中目标预测层的特征融合;使用K‑means聚类算法对训练集真实框高宽比进行聚类分析,调整候选框的高宽比,使候选框尺寸更适用于检测工件表面缺陷目标。本发明增加模型对多尺度缺陷目标的检测精度,提高对小缺陷目标的检测能力。

    一种基于改进的YOLOv5的织物瑕疵检测方法及装置

    公开(公告)号:CN116740050A

    公开(公告)日:2023-09-12

    申请号:CN202310857463.9

    申请日:2023-07-13

    申请人: 淮阴工学院

    摘要: 本发明公开一种基于改进的YOLOv5的织物瑕疵检测方法及装置,收集织物瑕疵数据集并对数据集进行预处理,对数据集进行数据增强以及均衡扩充;使用Kmeans++聚类算法对织物数据集瑕疵GT框聚类获得先验框;数据集按比例划分训练集和验证集;对模型进行改进,使用moscio‑9替换原来的moscio‑4进行在线数据增强,在特征提取的主干网络backbone中引入可变形卷积,提高模型的特征提取能力,在特征融合的颈部网络Neck中加入注意力机制,在检测网络的Head中引入Decoupled‑Detect;将织物瑕疵数据集通过搭建好的YOLOv5+DCN+BiFormer+Decoupled织物瑕疵检测网络进行训练获得模型预训练权重;将预处理后的待检测织物瑕疵图片输入最优权重模型进行瑕疵检测。本发明有助于提高织物瑕疵检测的准确性和检测模型的鲁棒性。