有效

目标检测模型的训练方法、目标检测方法、装置和设备

臧义华、沈礼权、马兴民、乐蕤、钟宋义、古润南
中国电子科技集团公司第十五研究所

摘要

本公开提供一种目标检测模型的训练方法、目标检测方法、装置和设备,包括:获取模型训练样本;基于模型训练样本对预设检测模型进行训练,得到初始检测模型;将模型训练样本输入初始检测模型的第一分支中,得到图像提取特征;基于模型训练样本和图像提取特征对预设压缩模型进行训练,得到图像压缩模型;将模型训练样本输入图像压缩模型中,得到图像压缩特征;将图像压缩特征输入初始检测模型的第二分支中,得到历史图像数据对应的目标检测结果;采用预设损失函数,基于历史图像数据对应的目标检测结果对第一分支和第二分支进行参数微调,得到目标检测模型。从而,有效提升目标检测模型的检测准确性。