基于深度学习算法的表面缺陷检测方法

    公开(公告)号:CN115205275A

    公开(公告)日:2022-10-18

    申请号:CN202210889914.2

    申请日:2022-07-27

    Abstract: 基于深度学习算法的表面缺陷检测方法,解决了现有采用深度学习算法进行表面缺陷检测时存在样本不均衡的问题,属于工业生产中检测技术领域。本发明包括:基于深度学习算法建立表面缺陷分类模型,利用初始训练样本集D1对表面缺陷分类模型进行训练;对D1中各类表面缺陷图像样本进行采样,获得均衡图像样本集D2,利用D2继续对表面缺陷分类模型进行训练,训练过程中,冻结表面缺陷分类模型中卷积层的参数,调整全连接层的参数,直至表面缺陷分类模型收敛;利用表面缺陷分类模型对待检测图像样本进行检测,计算弱监督定位信息;并进行阈值分割处理,利用二值化阈值,生成缺陷的位置掩膜,完成表面缺陷检测。

Patent Agency Ranking