一种基于轻量化分割模型的缺陷区域检测方法及装置

摘要

本申请提供一种基于轻量化分割模型的缺陷区域检测方法及装置。方法包括获取训练样本图像，并确定训练样本图像的缺陷区域，得到缺陷图像；根据预设的缺陷类别对缺陷图像进行标记，得到缺陷标记图像；将缺陷标记图像输入至轻量化分割模型中进行训练，得到训练好的轻量化分割模型；将待检测图像输入至训练好的轻量化分割模型中，检测并输出待检测图像中每个像素的缺陷类别概率值；将缺陷类别概率值相同的像素组合输出为缺陷检测区域。本申请中轻量化分割模型的解码部分用于恢复经过分类模型处理输出后待检测图像的分辨率，保证在提取到的高级特征图基础上，分辨率的准确性；并且解码部分的参数空间较小，分割模型更轻量，对缺陷区域的检测效率较高。