目标检测模型训练方法、检测方法及装置、设备、介质
摘要:
本公开提供了一种目标检测模型训练方法、检测方法及装置、设备、介质,该方法包括:利用学生模型基于无标注图像的语义特征对感兴趣区域的类进行预测,获得感兴趣区域的语义分类预测结果;利用教师模型和原型类集合对无标注图像中感兴趣区域进行语义相似性分类,获得感兴趣区域的语义伪标签;基于语义分类预测结果、语义伪标签和预设的语义对齐损失函数计算语义对齐损失;基于语义对齐损失更新学生模型中的模型参数,以及,利用学生模型中的模型参数更新教师模型中的模型参数,在满足终止条件的情况下,将教师模型确定为目标检测模型。本公开实施例提供的目标检测模型训练方法,可以降低训练成本,并提高模型的目标检测的精度。
0/0