一种基于半监督目标检测算法的钢材表面缺陷检测方法

专利主权项内容

1.一种基于半监督目标检测算法的钢材表面缺陷检测方法，其特征在于，包括以下步骤：数据采集阶段，采集若干钢材表面缺陷图像A；数据处理阶段，对部分所述钢材表面缺陷图像A进行缺陷标注，得到钢材表面缺陷图像B；模型构建阶段，构建教师模型和学生模型；模型训练阶段，包括：将所述钢材表面缺陷图像A输入所述教师模型，所述教师模型对所述钢材表面缺陷图像A中的钢材表面缺陷进行预测，并根据预测结果进行标注，生成带有伪标注的钢材表面缺陷图像C；将所述钢材表面缺陷图像B和所述钢材表面缺陷图像C输入所述学生模型，所述学生模型对所述钢材表面缺陷图像B和所述钢材表面缺陷图像C进行预测，计算所述钢材表面缺陷图像C的伪标注的损失，并更新所述学生模型；通过所述学生模型的网络参数更新所述教师模型的网络参数；检测阶段，将所述教师模型作为检测模型，用于钢材表面缺陷检测。