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

    公开(公告)号:CN112329873A

    公开(公告)日:2021-02-05

    申请号:CN202011261491.7

    申请日:2020-11-12

    Abstract: 本申请提供了一种目标检测模型的训练方法、目标检测方法及装置,方法包括:获取训练样本集;训练样本集中的样本为包含对象的图像;每个图像标注有对象对应的真实边界框;将训练样本集中的样本输入至预设神经网络模型中进行训练,得到对象对应的预测边界框;基于对象对应的真实边界框、预测边界框和预设损失函数计算模型误差损失,并基于模型误差损失调整模型参数;预设损失函数为基于边界框面积、边界框中心点距离和边界框宽高比确定的函数关系;当模型误差损失收敛时,终止模型训练,得到训练好的目标检测模型。本申请能够提高模型的收敛速度,提高模型的目标检测精准度。

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

    公开(公告)号:CN112329873B

    公开(公告)日:2024-09-06

    申请号:CN202011261491.7

    申请日:2020-11-12

    Abstract: 本申请提供了一种目标检测模型的训练方法、目标检测方法及装置,方法包括:获取训练样本集;训练样本集中的样本为包含对象的图像;每个图像标注有对象对应的真实边界框;将训练样本集中的样本输入至预设神经网络模型中进行训练,得到对象对应的预测边界框;基于对象对应的真实边界框、预测边界框和预设损失函数计算模型误差损失,并基于模型误差损失调整模型参数;预设损失函数为基于边界框面积、边界框中心点距离和边界框宽高比确定的函数关系;当模型误差损失收敛时,终止模型训练,得到训练好的目标检测模型。本申请能够提高模型的收敛速度,提高模型的目标检测精准度。

Patent Agency Ranking