一种目标样本标签的分配方法组成比例

本发明专利技术公开了图像检测领域的一种目标样本标签的分配方法，根据图像中的目标中心点确定初始标签区域；利用网络模型生成预选锚点，初步筛选出初始标签区域内的锚点；使用分类损失函数和位置回归损失函数计算每个锚点的分类损失值和位置回归损失值，将分类损失值和位置回归损失值按照相应权重累加，得到各锚点的总损失值；根据总损失值选取最终锚点，对选取的最终锚点进行标签分配，通过选取高置信度的部分点做为锚点，减少模糊锚点的标签分配，为图像中所有目标找到高置信度分配方式，同时，能减少检测系统训练时间，避免了添加额外的参数，有效提升检测系统的性能。有效提升检测系统的性能。有效提升检测系统的性能。

【技术实现步骤摘要】
一种目标样本标签的分配方法


[0001]本专利技术属于图像检测
，具体涉及目标样本标签的分配方法。

技术介绍

[0002]随着汽车电动化智能化进程的不断深入，国民的期望值提升，软件改变了汽车属性，自动驾驶的不断进化逐步解放了驾驶员的注意力，直至完全摆脱人工干预，车辆也由单纯的出行工具变成了移动私人空间。而自动驾驶的实现主要依靠各种传感器对车辆周围环境进行感知，通过目标检测，识别与跟踪等处理技术来评估危险性。
[0003]现阶段比较流行目标检测算法都通过预测一组预定义锚点的分类标签进行目标检测，例如一些方法采用固定中心点加角点的规则分配每个锚点所处的目标对象或背景，这种静态策略会存在一个问题，它会导致对于具有不同大小、形状或遮挡条件的目标对象的锚点划分边界会有所不同。对于单个目标的也就有很多不同置信度的锚点，远离目标中心的点置信度越低，由于目标的大小、重叠，框的贴合度等问题，经常出现一些模棱两可的锚点类别。

技术实现思路

[0004]本专利技术的目的在于提供一种目标样本标签的分配方法，通过选取高置信度的部分点做为锚点的标签，减少模糊锚点的标签分配，为图像中所有目标找到高置信度分配方式。
[0005]为达到上述目的，本专利技术所采用的技术方案是：
[0006]本专利技术提供了一种目标样本标签的分配方法，包括：
[0007]根据图像中的目标中心点确定初始标签区域；
[0008]提取图像的特征图信息，然后在特征图的每一特征点上生成预选锚点，初步筛选出初始标签区域内的锚点；
[0009]使用分类损失函数和位置回归损失函数计算初始标签区域内的每个锚点的分类损失值和位置回归损失值，将分类损失值和位置回归损失值按照相应权重累加，得到各锚点的总损失值；
[0010]根据各锚点的总损失值选取最终锚点；
[0011]对选取的最终锚点进行标签分配。
[0012]优选的，所述图像的两个对角点分别为(tl
x
，tl
y
)和(br
x
，br
y
)；所述初始标签区域的两个对角点分别为(ctl
x
，ctl
y
)和(cbr
x
，cbr
y
)；所述图像的两个对角点与所述初始标签区域的两个对角点的关系为：
[0013][0014]其中n为调整初始标签区域大小的参数。
[0015]优选的，所述分类损失函数为：
[0016][0017]其中，a是锚点预测的概率值；锚点的标记为目标时，y为1，否则，y为0；x是所属分类下各锚点,n'为所属分类下锚点的数量。
[0018]优选的，假设锚点的预测框为A，锚点的标注框为B，位置回归损失函数为：
[0019][0020][0021]其中IoU为预测框A与标注框B的交并比,代表A与B中心点之间的欧几里得距离，c为A与B之间最小外接矩形测对角线的长度，v表示A与B之间长宽比的距离，α是权重系数。
[0022]优选的，v的计算公式为：
[0023][0024]w
gt
为标注框的宽度，h
gt
为标注框的高度，w为预测框的宽度，h为预测框的高度。
[0025]优选的，权重系数α为：
[0026][0027]优选的，根据总损失值选取最终锚点，包括：根据总损失值对初始标签区域内的各锚点进行排序，由大到小选取设定数量a的锚点。
[0028]优选的，累加选取的设定数量a的锚点的总损失值并取整为数量b，作为重新选取锚点的数量，根据初始标签区域内各锚点的排序，由大到小选取数量b的锚点。
[0029]优选的，响应于有锚点重叠分配时，选择保留其中分类损失值最大的锚点，去除与分类损失值最大的锚点重叠的锚点。
[0030]与现有技术相比，本专利技术所达到的有益效果：
[0031]本专利技术根据图像中的目标中心点确定初始标签区域，获取所述初始标签区域的预
选锚点，将分类损失值和位置回归损失值按照相应权重累加，得到各锚点的总损失值；根据总损失值选取锚点，通过选取高置信度的部分点做为锚点，减少模糊锚点的标签分配，为图像中所有目标找到高置信度分配方式，同时，能减少检测系统训练时间，避免了添加额外的参数，有效提升检测系统的性能。
附图说明
[0032]图1是本专利技术实施例提供的图像标签分配示意图；
[0033]图2是本专利技术实施例提供的n为3时初始标签区域示意图；
[0034]图3是本专利技术实施例提供的n为5时初始标签区域示意图；
[0035]图4是本专利技术实施例提供的包含多个类别的图像。
[0036]图5是本专利技术实施例提供的一种目标样本标签的分配方法的流程图。
具体实施方式
[0037]下面结合附图对本专利技术作进一步描述。以下实施例仅用于更加清楚地说明本专利技术的技术方案，而不能以此来限制本专利技术的保护范围。
[0038]如图1
‑
5所示，一种目标样本标签的分配方法，包括：根据图像中的目标中心点确定初始标签区域；所述图像的左上角的点为(tl
x
，tl
y
)和右下角的点为(br
x
，br
y
)；所述初始标签区域的左上角的点为(ctl
x
，ctl
y
)和右下角的点为(cbr
x
，cbr
y
)；所述图像的两个对角点与所述初始标签区域的两个对角点的关系为：
[0039][0040]其中n为调整初始标签区域大小的参数。
[0041]利用网络模型提取图像的特征图信息，然后在特征图的每一特征点上生成预选锚点，初步筛选出初始标签区域内的锚点；
[0042]使用分类损失函数和位置回归损失函数计算初始标签区域内的每个锚点的分类损失值和位置回归损失值，将分类损失值和位置回归损失值按照相应权重累加，得到各锚点的总损失值；
[0043]所述分类损失函数CIOU Loss为：
[0044][0045]其中，a是锚点预测的概率值；锚点的标记为目标时，y为1，否则，y为0；x是所属分类下各锚点,n'为所属分类下锚点的数量。
[0046]优选的，假设锚点的预测框为A，锚点的标注框为B，位置回归损失函数为：
[0047][0048][0049]其中IoU为预测框A与标注框B的交并比，代表A与B中心点之间的欧几里得距离，c为A与B之间最小外接矩形测对角线的长度，v表示A与B之间长宽比的距离，α是权重系数。
[0050]v的计算公式为：
[0051][0052]w
gt
为标注框的宽度，h
gt
为标注框的高度，w为预测框的宽度，h为为预测框的高度；
[0053]权重系数α为：
[0054][0055]响应于有锚点重叠分配时，如图4所示，选取本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种目标样本标签的分配方法，其特征在于，包括：根据图像中的目标中心点确定初始标签区域；提取图像的特征图信息，然后在特征图的每一特征点上生成预选锚点，初步筛选出初始标签区域内的锚点；使用分类损失函数和位置回归损失函数计算初始标签区域内的每个锚点的分类损失值和位置回归损失值，将分类损失值和位置回归损失值按照相应权重累加，得到各锚点的总损失值；根据各锚点的总损失值选取最终锚点；对选取的最终锚点进行标签分配。2.根据权利要求1所述的一种目标样本标签的分配方法，其特征在于，所述图像的两个对角点分别为(tl
x
，tl
y
)和(br
x
，br
y
)；所述初始标签区域的两个对角点分别为(ctl
x
，ctl
y
)和(cbr
x
，cbr
y
)；所述图像的两个对角点与所述初始标签区域的两个对角点的关系为：其中n为调整初始标签区域大小的参数。3.根据权利要求1所述的一种目标样本标签的分配方法，其特征在于，所述分类损失函数为：其中，a是锚点预测的概率值；锚点的标记为目标时，y为1，否则，y为0；x是所属分类下各锚点,n'为所属分类下锚点的数量。4.根据权利要求1所述的一种目标样本标签的分配...

【专利技术属性】
技术研发人员：朱晓东，刘国清，季思文，
申请(专利权)人：南京佑驾科技有限公司，
类型：发明
国别省市：