有效

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

朱柯弘、张梓航、赵自然、顾建平、金颖康
同方威视技术股份有限公司
朱柯弘机构 暂无
技术领域 暂无
张梓航机构 暂无
技术领域 暂无
赵自然机构 暂无
技术领域 暂无
顾建平机构 暂无
技术领域 暂无
金颖康机构 暂无
技术领域 暂无

摘要

提供一种目标检测模型的训练方法和装置、目标检测方法。训练方法包括:获取训练样本集,包括第一和第二训练样本,第一和第二训练样本分别包括清晰数据和模糊数据;将第一训练样本输入目标检测模型;根据目标检测模型的输出值与第二训练样本,利用第一损失函数确定目标检测模型的第一损失值;将第一损失值输入高斯混合模型;利用高斯混合模型,构建模糊度矩阵,用于表示第一训练样本和第二训练样本中数据的模糊度;根据模糊度矩阵,构建模糊损失函数,模糊损失函数被构造为能够区分第一训练样本和第二训练样本中的清晰数据和模糊数据;根据模糊损失函数,构建第二损失函数;利用第一、第二训练样本集和第二损失函数,对目标检测模型进行重训练。