【技术实现步骤摘要】
本专利技术涉及图像识别领域，具体而言，涉及一种目标检测模型的确定方法及装置、存储介质。
技术介绍
1、近年来，深度学习发展迅猛，被广泛应用到计算机视觉、语音识别以及自然语言处理上。随着数据规模积累的足够大，场景越加复杂，对模型的精度也有越来越高的要求。目标检测作为计算机视觉应用的主要探索之一，其模型和训练方法也得到一次次的改进和研发。目标检测训练方案中标签分配策略（label assigner）对训练精度的影响非常大，一个好的标签分配策略不仅能大幅度提升模型精度，还可以提升训练速度，加速模型收敛。而目前主流的标签分配策略有基于侯选框（bbox）和真实框(gt_box)的交并比来确定的，有基于bbox的中心点是否落在gt_box内来确定的，有通过统计学方案确定的如atss_assigner，也有基于模型训练不断优化的如simota_assigner。但目前的研究中，均集中在如何改进label assigner的策略来提升检测模型的精度，改进效果并不显著。
2、针对相关技术，现有检测模型对图片进行目标检测的准确度较低的问题，目前尚未提出有效的解决方案。
3、因此，有必要对相关技术予以改良以克服相关技术中的所述缺陷。
技术实现思路
1、本专利技术实施例提供了一种目标检测模型的确定方法及装置、存储介质，以至少解决现有检测模型对图片进行目标检测的准确度较低的问题。
2、根据本专利技术实施例的一方面，提供一种目标检测模型的确定方法，包括：获取训练样本集合，其
3、在一个示例性的实施例中，使用所述训练样本集合对待训练的检测模型进行训练，包括：通过以下步骤执行第i轮训练，其中，i为大于或等于1的正整数，经过第0轮训练的检测模型是未经过训练的所述待训练的检测模型：获取第i-1轮训练得到的目标网络对样本图片进行检测后得到的预测框集合；使用所述第i-1轮训练得到的n个标签分配器分别对所述第i轮检测出的预测框集合进行标签分配，得到n个标签分配预测结果，其中，每个标签分配预测结果包括所述预测框集合中每个预测框的标签；根据所述n个标签分配预测结果确定所述第i轮训练的目标分类损失函数的取值以及目标定位损失函数的取值，其中，所述目标损失函数包括所述目标分类损失函数和所述目标定位损失函数；所述n个标签分配器在得到所述n个标签分配预测结果的过程中产生了所述n组损失；在所述第i轮训练的所述目标定位损失函数的取值和所述目标分类损失函数的取值均满足对应的收敛条件的情况下，结束训练；在所述第i轮训练的所述目标定位损失函数的取值不满足对应的收敛条件、或者所述目标分类损失函数的取值不满足对应的收敛条件的情况下，调整经过第i-1轮训练得到的检测模型中的参数，得到经过所述第i轮训练得到的检测模型。
4、在一个示例性的实施例中，根据所述n个标签分配预测结果确定所述第i轮训练的目标分类损失函数的取值，包括：根据n个标签分配预测结果确定n个矩阵，其中，所述n个矩阵中的第j个矩阵用于表示第j个标签分配器产生的标签分配预测损失，所述第j个矩阵中的元素值包括第j个标签分配器为所述预测框集合中每个预测框进行标签分配预测产生的分配预测损失；根据所述n个矩阵确定所述第i轮训练的目标分类损失函数的值。
5、在一个示例性的实施例中，根据n个标签分配预测结果确定n个矩阵，包括：通过以下方式得到n个矩阵中的第j个矩阵，以得到所述n个矩阵：将n个标签分配预测结果中的第j个标签分配预测结果和与所述第j个标签分配预测结果对应的标签分配结果输入至分类损失函数，得到第j个矩阵；其中，所述第j个标签分配预测结果对应的标签分配结果是根据所述第j个标签分配器的标签分配策略、所述样本图片的多个标注框和每个标注框对应的标签对所述第i轮检测出的预测框集合进行标签分配的结果。
6、在一个示例性的实施例中，根据所述n个矩阵确定所述第i轮训练的目标分类损失函数的值，包括：从所述n个矩阵中确定目标矩阵；根据所述n个矩阵中除所述目标矩阵以外的n-1个矩阵确定目标权重矩阵，其中，目标权重矩阵中的元素值包括所述目标矩阵中每个预测框对应的分配预测损失的权重值；将所述目标权重矩阵和所述目标矩阵进行矩阵对位相乘运算，得到第i轮训练的目标分类损失函数的取值。
7、在一个示例性的实施例中，根据所述n个矩阵中除所述目标矩阵以外的n-1个矩阵确定目标权重矩阵，包括：将所述n-1个矩阵进行加法运算，得到第一中间矩阵；将所述第一中间矩阵进行归一化处理，得到第二中间矩阵；将预设矩阵与所述第二中间矩阵进行减法运算，得到目标权重矩阵，其中，所述预设矩阵与所述第二中间矩阵具有相同的行列数、且所述预设矩阵的每个元素值等于1。
8、在一个示例性的实施例中，根据所述n个标签分配预测结果确定所述第i轮训练的目标定位损失函数的取值，包括：根据所述n个标签分配预测结果确定n个标签分配器产生的n个定位损失；根据所述n个定位损失确定所述第i轮训练的目标定位损失函数的取值。
9、在一个示例性的实施例中，根据n个标签分配预测结果确定n个标签分配器产生的n个定位损失，包括：通过以下方式得到n个定位损失中的第j个定位损失，以得到n个定位损失：根据所述n个标签分配预测结果中的第j个标签分配预测结果确定第j组预测框，其中，所述第j组预测框中的每个预测框对应的标签不为背景标签；使用定位损失函数确定所述第j组预测框对应的位置信息与所述多个标注框对应的位置信息之间的定位损失，得到第j个定位损失。
10、在一个示例性的实施例中，将结束训练时的待训练的检测模型确定为目标检测模型之后，所述方法还包括：获取待进行目标检测的目标图片；通过所述目标检测模型对所述目标图片进行目标检测，以在所述目标图片中确定多个检测框以及所述多个检测框中每个检测框的标签。
11、根据本专利技术实施例的另一方面，还提供了一种目标检测模型的确定装置，包括：获取模块，用于获取训练样本集合，其中，所述训练样本集合中的每个训练样本包括：样本图片、所述样本图片的多个标注框和每个标注框对应的标签；训练模块，用于使用所述训练样本集合对待训练的检测模型进行训练，直到所述待训练的检测模型对应的目标损失函数的取值满足预设的收敛条件，其中，所述待训练的检测模型包括：用于对所述样本图片进行目标检测的目标网络和用于分别对所述目标网络检测出的预测框集合进行标签分配预测的n个标签分配器；n为大于1的正整数，所述目标损失函数的本文档来自技高网...
【技术保护点】
1.一种目标检测模型的确定方法，其特征在于，包括：
2.根据权利要求1所述的方法，其特征在于，使用所述训练样本集合对待训练的检测模型进行训练，包括：
3.根据权利要求2所述的方法，其特征在于，根据所述N个标签分配预测结果确定所述第i轮训练的目标分类损失函数的取值，包括：
4.根据权利要求3所述的方法，其特征在于，根据N个标签分配预测结果确定N个矩阵，包括：
5.根据权利要求3所述的方法，其特征在于，根据所述N个矩阵确定所述第i轮训练的目标分类损失函数的值，包括：
6.根据权利要求5所述的方法，其特征在于，根据所述N个矩阵中除所述目标矩阵以外的N-1个矩阵确定目标权重矩阵，包括：
7.根据权利要求2所述的方法，其特征在于，根据所述N个标签分配预测结果确定所述第i轮训练的目标定位损失函数的取值，包括：
8.根据权利要求7所述的方法，其特征在于，根据N个标签分配预测结果确定N个标签分配器产生的N个定位损失，包括：
9.根据权利要求1所述的方法，其特征在于，将结束训练时的待训练的检测模型确定为
10.一种目标检测模型的确定装置，其特征在于，包括：
11.一种计算机可读的存储介质，其特征在于，所述计算机可读的存储介质包括存储的程序，其中，所述程序运行时执行权利要求1至9任一项中所述的方法。
12.一种电子装置，包括存储器和处理器，其特征在于，所述存储器中存储有计算机程序，所述处理器被设置为通过所述计算机程序执行权利要求1至9任一项中所述的方法。
...【技术特征摘要】
1.一种目标检测模型的确定方法，其特征在于，包括：
2.根据权利要求1所述的方法，其特征在于，使用所述训练样本集合对待训练的检测模型进行训练，包括：
3.根据权利要求2所述的方法，其特征在于，根据所述n个标签分配预测结果确定所述第i轮训练的目标分类损失函数的取值，包括：
4.根据权利要求3所述的方法，其特征在于，根据n个标签分配预测结果确定n个矩阵，包括：
5.根据权利要求3所述的方法，其特征在于，根据所述n个矩阵确定所述第i轮训练的目标分类损失函数的值，包括：
6.根据权利要求5所述的方法，其特征在于，根据所述n个矩阵中除所述目标矩阵以外的n-1个矩阵确定目标权重矩阵，包括：
7.根据权利要求2所述的方法，其特征在于，根据所述n...
【专利技术属性】
技术研发人员：林亦宁，杨德城，
申请(专利权)人：上海闪马智能科技有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。