一种缺陷检测模型的训练方法和相关装置
摘要:
本申请公开一种缺陷检测模型的训练方法和相关装置,获取训练数据集和元数据集,训练数据集中包括第一产品图像和第一产品图像的噪声标签,元数据集中包括第二产品图像和第二产品图像的干净标签。基于第二产品图像和第二产品图像的干净标签构建目标损失函数,并以目标损失函数最小为目标对标签纠正网络进行训练,在以目标损失函数最小为目标对标签纠正网络进行训练的过程中,利用训练得到的标签纠正网络对噪声标签进行不断的纠正得到目标标签,目标标签相对于噪声标签来说更加准确,故可以根据第一产品图像和更加准确的目标标签训练得到缺陷检测模型。从而提高缺陷检测模型训练的准确性,进而提高缺陷检测模型的准确性、可靠性。
公开/授权文献
0/0