一种基于改进图注意力机制的密封钉缺陷检测的方法、系统、计算机可读存储介质及计算机程序产品

本发明涉及缺陷检测技术领域，公开了一种基于改进图注意力机制的密封钉缺陷检测的方法、系统、计算机可读存储介质及计算机程序产品。所述方法包括：构建用于分割缺陷区域的语义分割网络；用向量注意力算子聚合局部特征，生成相关系数矩阵，并根据空间距离关系对相关系数矩阵进行约束，获取语义分割网络的注意力权重矩阵；为边缘卷积运算过程附加法向量特征编码，获取语义分割网络的边缘卷积算子；基于所述语义分割网络构建语义分割模型，并利用预先构建的密封钉缺陷训练集，对语义分割模型进行训练，获取语义分割模型的网络参数。本发明的语义分割模型能精确分割缺陷区域，识别传统算法无法识别的潜在的缺陷，有效地提高了缺陷检测的准确率。