一种聚丙烯薄膜缺陷检测模型的训练方法、装置及聚丙烯薄膜缺陷检测方法

本发明公开了一种聚丙烯薄膜缺陷检测模型的训练方法、装置及聚丙烯薄膜缺陷检测方法，涉及工业薄膜检测领域，训练方法包括：获取薄膜训练图像，通过伪标签模型生成伪标签；基于薄膜训练图像和伪标签对聚丙烯薄膜缺陷检测模型进行迭代训练；其中，伪标签模型的构建过程为：获取标注数据集和未标注数据集，并输入至待训练的伪标签模型，通过学生网络根据标注数据集生成预测值，并计算有监督损失；根据学生网络参数确定教师网络参数；通过学生网络和教师网络分别根据未标注数据集生成预测值，并计算一致性损失；各损失值相加确定总损失；不断调整学生网络参数直至总损失收敛，得到伪标签模型的网络参数。通过实施本发明，能够降低标注成本。