【技术实现步骤摘要】
本申请属于计算机视觉，尤其涉及一种标签分配方法、标签分配装置、电子设备及存储介质。
技术介绍
1、标签分配为将目标检测模型前向传播中输出的预测框与样本标注的真实框进行匹配的过程，标签分配将预测框划分为正样本和负样本，正负样本的均衡性决定了目标检测模型的检测准确性。
2、目前，主要根据预测框与真实框的交并比(intersection over union，iou)，对预测框进行标签分配，容易将置信度较高的小预测框误划分为负样本，正负样本数量将出现严重不均衡，导致目标检测模型的检测准确性降低。
技术实现思路
1、有鉴于此，本申请实施例提供了一种标签分配方法、标签分配装置、电子设备及存储介质，以克服以上现有技术的问题。
2、第一方面，本申请实施例提供了一种标签分配方法，包括：获取样本图像的多个目标预测框信息，每个目标预测框信息包括一个目标预测框及一个目标置信度；确定每个目标预测框与所述样本图像的真实框的一个相似度，以得到多个相似度，每个相似度基于所述一个目标预测框对应的一...
【技术保护点】
1.一种标签分配方法，其特征在于，包括：
2.根据权利要求1所述的标签分配方法，其特征在于，所述确定每个目标预测框与所述样本图像的真实框的一个相似度，包括：
3.根据权利要求2所述的标签分配方法，其特征在于，所述根据所述每个目标预测框对应的所述一个目标交并比及所述目标置信度，确定所述一个相似度，包括：
4.根据权利要求3所述的标签分配方法，其特征在于，所述根据每个第一坐标与所述第二坐标，确定所述一个相似度，包括：
5.根据权利要求1所述的标签分配方法，其特征在于，所述获取样本图像的多个目标预测框信息，包括：
...
【技术特征摘要】
1.一种标签分配方法，其特征在于，包括：
2.根据权利要求1所述的标签分配方法，其特征在于，所述确定每个目标预测框与所述样本图像的真实框的一个相似度，包括：
3.根据权利要求2所述的标签分配方法，其特征在于，所述根据所述每个目标预测框对应的所述一个目标交并比及所述目标置信度，确定所述一个相似度，包括：
4.根据权利要求3所述的标签分配方法，其特征在于，所述根据每个第一坐标与所述第二坐标，确定所述一个相似度，包括：
5.根据权利要求1所述的标签分配方法，其特征在于，所述获取样本图像的多个目标预测框信息，包括：
【专利技术属性】
技术研发人员：蒋海辉，孔德深，梁玮琪，殷家汇，史志强，李光志，
申请(专利权)人：长园深瑞继保自动化有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。