【技术实现步骤摘要】
用于工业外观缺陷检测的人工智能分割分类算法
[0001]本专利技术涉及工业计算机视觉
,特别涉及用于工业外观缺陷检测的人工智能分割分类算法。
技术介绍
[0002]目前的工业外观缺陷检测算法主要是基于传统图像算法,存在问题有:可应用场景有限(基于正负样本库的图像差值对比,提供公差类检测服务,无法对划伤、异物、缺损等无规则缺陷保存标准样本)、检测精度低(无法提取各种缺陷内在特征,当待检品与标准样本有出入时,可能会误判)、普适性差(针对产品改良和规格升级条件下,无法对海量标准样本进行算法重构)、天花板明显(当缺陷与背景的边缘特征很不明显时,传统图像算法所表现的天花板越来越明显),国内目前还没有自主研发的用于外观缺陷检测的人工智能分割分类算法。
[0003]针对工业领域缺陷检测业务需求、传统图像算法应用局限、本算法分割和分类两个层面展开研发,第一个层面是分割,保证外观缺陷检出力,满足业务需求,第二个层面是分类,对分割检出的结果进行复盘,保证误判在工程可接受范围内,本算法的缺陷检出率趋近于100%、漏检率趋近于0%。直 ...
【技术保护点】
【技术特征摘要】
1.用于工业外观缺陷检测的人工智能分割分类算法,其特征在于:包括以下具体步骤,首先通过样本的轮廓标注得到训练数据集,然后构建语义分割格式的训练集,完成训练的预处理,然后构建语义分割模型进行模型的训练,最后利用训练的分割模型进行在线推理检测出缺陷的精确位置;其中语义分割首先需要人工标注待检测目标的边缘轮廓信息,包含精细的边缘轮廓以及轮廓内包含的目标类别;然后把标注好的样本集按类别随机打乱,然后组装成语义分割可以理解的数据格式,即每个标注区域内用该类别对应的像素值填充,把目标与目标之间的标签信息相互隔离,组装成语义分割可以训练的数据格式;之后通过对整幅图像进行OK和NG打标签得到训练数据集,然后构建分类格式的训练集,完成训练的预处理,通过对整幅图像进行OK和NG打标签得到训练数据集,然后构建分类格式的训练集,完成训练的预处理,图像分类首先需要人工对每幅图像打标签。2.根据权利要求1所述的用于工业外观缺陷检测的人工智能分割分类算法,其特征在于:所述构建语义分割模型选择UNet模型,所述语义分割模型还包含四层下采样作为编码模块和四层上采样作为解码模块,编码模块和解码模块左右完全对称,且相对应的层之间进行浅层和深层的特征融合,融合完成后得到一个跟输入图像相同大小的预测概率图,最后计算输入图像与上一步得到的概率图之间的损失,通过损失的反向传播来训练语义分割模型;其中图像卷积公式:W
out
=(W
in
+2*P
‑
F)/S+T式中,W
in
为输入图像的宽,F为滤波器的大小,S为移动步长,P为卷积时扩展的像素数,T为实验参数。3.根据权利要求2所述的用于工业外观缺陷检测的人工智能分割分类算法,其特征在于:其中损失函数包括DiceLoss、FocalLoss和带权的交叉熵损失,其中...
【专利技术属性】
技术研发人员:孙开,陈绍林,
申请(专利权)人:苏州数算人工智能科技有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。