摘要
本发明提供了一种目标检测方法、装置及设备,涉及领域自适应领域。包括:对目标域图像样本进行不同处理后分别输入源域的预训练的教师模型和目标域的待训练的学生模型,得到教师编码特征、教师解码特征、学生编码特征和学生解码特征;基于学生类别预测结果得到不确定性权重;基于教师编码特征、教师解码特征、学生编码特征以及不确定性权重得到特征蒸馏损失;基于伪标签、学生类别预测结果、学生边界框预测结果、学生解码特征和多种类别的记忆库得到对比学习损失;基于特征蒸馏损失和对比学习损失对进行模型更新,得到训练完毕的学生模型,以对目标域的待检测图像进行目标检测,实现将DETR检测模型应用到无源场景下的领域自适应目标检测。
技术关键词
学生
编码特征
教师
蒸馏
记忆
多尺度
样本
标签
检测损失
解码器
模型更新
编码器
融合特征
输入模块
模型训练模块
图像处理模块
处理器
指数