金融界2024年3月27日消息,据国家知识产权局公告,贝壳找房(北京)科技有限公司申请一项名为“半监督目标检测模型的训练方法、存储介质和电子设备“,公开号CN117765245A,申请日期为2023年12月。

专利摘要显示,公开了一种半监督目标检测模型的训练方法、存储介质和电子设备,该方法包括:将经过弱数据增强处理的未标注数据输入第一模型,经第一模型输出至少一个伪真实框;将经过强数据增强处理的未标注数据输入第二模型,经第二模型输出至少一个第一预测框;利用当前训练轮次的各类别的置信度阈值,确定至少一个伪真实框中的至少一个目标伪真实框;确定与各目标伪真实框匹配的第一预测框;根据正样本、负样本及匹配的目标伪真实框,计算无监督损失函数值;执行梯度反向传播,更新第二模型的模型参数;响应于确定模型训练过程未完成,执行下一轮次的训练过程。本公开实施例公开了将半监督目标检测方法应用到DETR检测器的实现方式。

本文源自金融界