有效

一种半监督目标检测方法与系统

张文彬、王东林、张军民、崔隽峰、席晓强、李海冰、刘晨、王昌幼、周晋、张国梁、吴鹏、张屹
国网江苏省电力有限公司苏州供电分公司

摘要

一种半监督目标检测方法,用于检测待测样本图像的缺陷检测框与缺陷类别,包括:获取样本图像,包括:缺陷样本图像与待检测样本图像;根据带标签的缺陷样本图像,完成对Teacher模型和目标检测模型的预训练;根据目标检测模型与Student模型,获取每一个缺陷样本图像对应的Teacher模型的检测框和分类结果与Student模型的检测框和分类结果,并计算分类结果之间的距离L<Sub>diff</Sub>;判断距离L<Sub>diff</Sub>与损失L<Sub>s</Sub>之和L是否收敛,若收敛,则表明Teacher模型和Student模型训练完成;否则继续迭代执行;将待测样本图像输入到Teacher模型和Student模型中,输出待测样本图像的缺陷检测框与缺陷类别。该发明联合训练有标签以及无标签样本数据,提升目标检测模型精确率。