一种基于多点位双分支模型的工业视觉缺陷检测方法

摘要

本发明公开了一种基于多点位双分支模型的工业视觉缺陷检测方法，涉及人工智能技术领域。本发明包括如下步骤：根据点位划分数据：根据点位信息对数据进行划分，得到多个正样本池和负样本池，并进行样本对齐；多点位采样策略：根据多点位采样策略分别从正样本池和负样本池中挑选正样本对和负样本对；特征提取：以ResNet‑50模型作为特征提取器，进行输入数据的特征提取；双分支预测：将提取的特征输入到包含注意力机制的若干全连接层，进行缺陷预测。本发明通过利用点位信息，可以有效地划分数据集，将复杂多样的数据集转换为若干点位内的子数据集，从而降低了网络学习难度，提升了模型的缺陷识别能力。