一种缺陷识别模型训练方法、缺陷识别方法、装置和设备

本发明公开了一种缺陷识别模型训练方法、缺陷识别方法、装置和设备，应用于缺陷检测领域，包括：对配电柜缺陷图像数据集中每个配电柜缺陷图像中的缺陷进行标注，得到小样本数据集；获取基于空间注意力机制、通道注意力机制和基础模型构建的预训练缺陷识别模型；基础模型为基于大样本数据集进行训练得到的模型；利用小样本数据集对预训练缺陷识别模型进行训练，得到目标缺陷识别模型。本申请中的目标缺陷识别模型可以基于训练好的空间注意力机制、通道注意力机制分别模拟空间维度和通道维度中的语义相互依赖性，解决现有的智能识别算法在小样本多缺陷识别任务下难以捕捉数据分布的全貌的问题，从而提高了配电柜缺陷识别的准确性。