缺陷检测模型的构建方法、缺陷检测方法、装置及电子设备

本申请涉及缺陷检测技术领域，公开一种缺陷检测模型的构建方法、缺陷检测方法、装置及电子设备。构建方法包括：将样本数据输入至YOLO特征提取网络，获得训练特征图；采用训练特征图对预先构建好的缺陷识别模型进行训练；缺陷识别模型包括图像分类模块、目标检测模块和目标分割模块；将YOLO特征提取网络与训练好的缺陷识别模型融合，获得缺陷检测模型；其中，图像分类模块用于对输入的特征图进行分类，输出不含有缺陷的正常图像或含有缺陷的异常图像，目标检测模块用于输出目标缺陷的边界框位置，目标分割模块用于输出目标缺陷的像素级分割图谱。该方法的技术效果说明。本申请，能够提升缺陷检测模型对小目标的缺陷检测精度，提升检测性能。