[发明专利]基于顶点距离与交并比多重优化目标检测器的方法与系统

权利要求书

1.一种基于顶点距离与交并比多重优化目标检测器的方法，其特征在于，包括：

获取锚与标注框的交并比IOU，所述标注框为矩形标注框；

基于锚与标注框对应的四个顶点之间的归一化距离系数distance，修正所述交并比IOU，得到修正交并比P-IOU；

基于修正交并比P-IOU重新界定正负样本；以及

基于重新界定正负样本分类训练检测器；

其中，所述锚与标注框对应的四个顶点之间的归一化距离系数的获取包括以下过程：

获取锚与标注框相对应的四个点的直线距离p1、p2、p3和p4，其中锚与标注框均在同一平面内；

分别计算所有标注框的宽高平均值h_w；以及

获取每个锚与标注框之间的归一化距离系数distance；

所述归一化距离系数distance根据以下公式确定：

。

2.根据权利要求1所述的基于顶点距离与交并比多重优化目标检测器的方法，其特征在于，根据归一化距离系数distance与交并比IOU的乘积修正锚与所有标注框的交并比IOU，得到修正交并比P-IOU。

3.根据权利要求1或2所述的基于顶点距离与交并比多重优化目标检测器的方法，其特征在于，基于修正交并比重新界定正负样本的处理包括以下过程：

保留每个锚对应的修正交并比的最大值以及对应标注框的类别，并且如果修正交并比P-IOU的最大值为0，则判定该锚类别为背景，该锚判定为负样本；

将所有锚对应的修正交并比P-IOU的最大值与训练预设的阈值进行比对，对于锚对应的修正交并比P-IOU的最大值大于训练预设的阈值的，保留此锚匹配的标注框的类别，并判定该锚为正样本；对于锚对应的修正交并比P-IOU的最大值小于或者等于训练预设的阈值的，将该锚匹配的标注框的类别修改成背景，判定该锚为负样本。

4.根据权利要求1所述的基于顶点距离与交并比多重优化目标检测器的方法，其特征在于，以重新界定后的正样本和负样本作为训练集，基于OHEM算法分类训练检测器。

5.一种基于顶点距离与交并比多重优化目标检测器的装置，其特征在于，包括：

用于获取锚与标注框的交并比IOU的获取模块；

用于基于锚与标注框对应的四个顶点之间的归一化距离系数distance，修正所述交并比IOU，得到修正交并比P-IOU的交并比修正模块；

用于基于修正交并比P-IOU重新界定正负样本的样本分类模块；以及

用于基于重新界定正负样本分类训练检测器的训练模块；

其中，所述交并比修正模块包括归一化距离系数获取模块以及修正模块，其中：

所述归一化距离系数获取模块被设置成根据以下过程获取归一化距离系数：

获取锚与标注框相对应的四个点的直线距离p1、p2、p3和p4，其中锚与标注框均在同一平面内；

分别计算所有标注框的宽高平均值h_w；以及

获取每个锚与标注框之间的归一化距离系数distance；

所述修正模块用于根据归一化距离系数distance与交并比IOU的乘积修正锚与所有标注框的交并比IOU，得到修正交并比P-IOU；

所述归一化距离系数distance根据以下公式确定：

。

6.根据权利要求5所述的基于顶点距离与交并比多重优化目标检测器的装置，其特征在于，所述样本分类模块被设置以下述方式重新界定正负样本：

保留每个锚对应的修正交并比P-IOU的最大值以及对应标注框的类别，并且如果修正交并比P-IOU的最大值为0，则判定该锚类别为背景，该锚判定为负样本；

将所有锚对应的修正交并比P-IOU的最大值与训练预设的阈值进行比对，对于锚对应的修正交并比P-IOU的最大值大于训练预设的阈值的，保留此锚匹配的标注框的类别，并判定该锚为正样本；对于锚对应的修正交并比P-IOU的最大值小于或者等于训练预设的阈值的，将该锚匹配的标注框的类别修改成背景，判定该锚为负样本。