一种基于混合注意力机制和类别特征优化的神经网络及缺陷检测方法

本发明提供一种基于混合注意力机制和类别特征优化的神经网络及缺陷检测方法，涉及缺陷检测技术领域，所述神经网络包括特征提取网络、Transformer网络、十字交叉注意力网络、特征融合网络、类别特征优化网络以及全连接层；所述特征提取网络用于特征图构建，所述Transformer网络用于特征图依赖关系构建；所述十字交叉注意力网络用于特征图全局像素点关联；所述特征融合网络用于特征图的特征融合；所述类别特征优化网络用于特征图的特征信息平滑处理，所述全连接层用于特征图的缺陷类别预测。本发明采用结构优化的神经网络进行面板缺陷检测，能够解决现有神经网络进行面板缺陷检测时容易出现漏检和误检的问题。