[发明专利]一种目标标注方法和一种目标标注装置

权利要求书

1.一种目标标注方法，其特征在于，该方法包括如下步骤：

使用目标检测模型检测包含目标的图片集合，得到目标的预测框，所述图片集合中包含已标注的图片和未标注的图片；

从所述预测框中，筛选出与已标注目标重合程度低于预设值或缺少已标注目标的预测框；

根据筛选出的预测框的面积和相对于面积的像素均值，查找出各预测框最接近的目标类别，若最接近的目标类别与该预测框的预测类别一致，则将该预测框的信息写入标注文件，

所述根据筛选出的预测框的面积和相对于面积的像素均值，查找出各预测框最接近的目标类别，包括：

统计已标注的图片中各已标注目标的标注框面积以及相对于面积的像素均值，获得各目标类别的中心点；

根据预测框的面积和相对于面积的像素均值，计算与各预测框最匹配的目标类别中心点，该中心点所对应的目标类别即为各预测框最接近的目标类别。

2.根据权利要求1所述的目标标注方法，其特征在于，筛选出与已标注目标重合程度低于预设值的预测框，包括：

计算预测框与已标注目标的标注框之间的CIoU，若CIoU小于预设值，则确定该预测框为与已标注目标重合程度低于预设值的预测框。

3.根据权利要求1所述的目标标注方法，其特征在于，所述根据预测框的面积和相对于面积的像素均值，计算与各预测框最匹配的目标类别中心点，包括：

以面积为横坐标、像素均值为纵坐标绘制面积-像素均值散点图；

在所述面积-像素均值散点图上，标注各预测框坐标以及各目标类别中心点坐标，计算各预测框坐标与各目标类型中心点坐标之间的欧式距离，所得欧式距离最小的目标类别中心点即为与该预测框最匹配的目标类别中心点。

4.根据权利要求1所述的目标标注方法，其特征在于，所述统计已标注的图片中各已标注目标的标注框面积以及相对于面积的像素均值，获得各目标类别的中心点，包括：

以一目标类别所有已标注目标的标注框面积的平均值作为该目标类别中心点的面积，以一目标类别所有已标注目标的标注框像素均值的平均值作为该目标类别中心点的像素均值。

5.根据权利要求1所述的目标标注方法，其特征在于，所述未标注的图片，包括全无标注的图片以及标注不完全的图片。

6.根据权利要求1所述的目标标注方法，其特征在于，该方法还包括：对于预测框中与已标注目标重合程度不小于预设值的预测框，将其预测框信息直接写入标注文件。

7.根据权利要求1所述的目标标注方法，其特征在于，所述方法还包括：

利用已标注的训练图片集合训练所述目标检测模型，所述训练图片集合均匀含盖所有类别的目标，所述目标检测模型为yolov4模型。

8.根据权利要求1所述的目标标注方法，其特征在于，所述预测框的信息包括：目标的中心点横坐标、中心点纵坐标、长、宽和预测类别。

9.一种目标标注装置，其特征在于，包括：

处理器；以及

被安排成存储计算机可执行指令的存储器，所述可执行指令在被执行时使所述处理器执行如权利要求1～8之任一所述目标标注方法。