基于交互注意力与卷积神经网络的瓷砖表面缺陷检测方法
摘要:
本发明公开了一种基于交互注意力与卷积神经网络的瓷砖表面缺陷检测方法,对收集到的瓷砖缺陷图像进行变换、特征提取,随后进行对齐,获得差分图像,结合缺陷图像与差分图像构成数据集,随后进行缺陷标注,按比例划分训练集测试集。确定优化方法及损失函数,将所构建的瓷砖表面缺陷检测神经网络以端到端的方式进行训练,获得训练好的检测神经网络对待检测瓷砖图像进行缺陷类别与缺陷位置预测,随后使用非极大值抑制获得最终结果。本发明提高了瓷砖缺陷检测的精度;本方法对瓷砖表面缺陷进行检测的方法流程可以应用到其他类型的表面缺陷检测框架中提高检测精度,通用性强。
0/0