目标检测模型的确定方法及装置、存储介质制造方法及图纸

技术编号:40002204 阅读:23 留言:0更新日期:2024-01-09 03:56
本发明专利技术公开了一种目标检测模型的确定方法及装置、存储介质,其中,上述方法包括:获取训练样本集合,其中,训练样本集合中的每个训练样本包括:样本图片、样本图片的多个标注框和每个标注框对应的标签;使用训练样本集合对待训练的检测模型进行训练,直到待训练的检测模型对应的目标损失函数的取值满足预设的收敛条件,其中,待训练的检测模型包括:用于对样本图片进行目标检测的目标网络和用于分别对目标网络检测出的预测框集合进行标签分配预测的N个标签分配器;N为大于1的正整数,目标损失函数的取值根据N个标签分配器产生的N组损失共同确定;将结束训练时的待训练的检测模型确定为目标检测模型。

【技术实现步骤摘要】

本专利技术涉及图像识别领域,具体而言,涉及一种目标检测模型的确定方法及装置、存储介质


技术介绍

1、近年来,深度学习发展迅猛,被广泛应用到计算机视觉、语音识别以及自然语言处理上。随着数据规模积累的足够大,场景越加复杂,对模型的精度也有越来越高的要求。目标检测作为计算机视觉应用的主要探索之一,其模型和训练方法也得到一次次的改进和研发。目标检测训练方案中标签分配策略(label assigner)对训练精度的影响非常大,一个好的标签分配策略不仅能大幅度提升模型精度,还可以提升训练速度,加速模型收敛。而目前主流的标签分配策略有基于侯选框(bbox)和真实框(gt_box)的交并比来确定的,有基于bbox的中心点是否落在gt_box内来确定的,有通过统计学方案确定的如atss_assigner,也有基于模型训练不断优化的如simota_assigner。但目前的研究中,均集中在如何改进label assigner的策略来提升检测模型的精度,改进效果并不显著。

2、针对相关技术,现有检测模型对图片进行目标检测的准确度较低的问题,目前尚未提出有效的解决方本文档来自技高网...

【技术保护点】

1.一种目标检测模型的确定方法,其特征在于,包括:

2.根据权利要求1所述的方法,其特征在于,使用所述训练样本集合对待训练的检测模型进行训练,包括:

3.根据权利要求2所述的方法,其特征在于,根据所述N个标签分配预测结果确定所述第i轮训练的目标分类损失函数的取值,包括:

4.根据权利要求3所述的方法,其特征在于,根据N个标签分配预测结果确定N个矩阵,包括:

5.根据权利要求3所述的方法,其特征在于,根据所述N个矩阵确定所述第i轮训练的目标分类损失函数的值,包括:

6.根据权利要求5所述的方法,其特征在于,根据所述N个矩阵中除所述...

【技术特征摘要】

1.一种目标检测模型的确定方法,其特征在于,包括:

2.根据权利要求1所述的方法,其特征在于,使用所述训练样本集合对待训练的检测模型进行训练,包括:

3.根据权利要求2所述的方法,其特征在于,根据所述n个标签分配预测结果确定所述第i轮训练的目标分类损失函数的取值,包括:

4.根据权利要求3所述的方法,其特征在于,根据n个标签分配预测结果确定n个矩阵,包括:

5.根据权利要求3所述的方法,其特征在于,根据所述n个矩阵确定所述第i轮训练的目标分类损失函数的值,包括:

6.根据权利要求5所述的方法,其特征在于,根据所述n个矩阵中除所述目标矩阵以外的n-1个矩阵确定目标权重矩阵,包括:

7.根据权利要求2所述的方法,其特征在于,根据所述n...

【专利技术属性】
技术研发人员:林亦宁杨德城
申请(专利权)人:上海闪马智能科技有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1