[发明专利]图像数据处理方法及装置、目标检测模型训练方法及装置

权利要求书

1.一种图像数据处理方法，其中，所述图像数据处理方法包括：

获取第一图像集，所述第一图像集包括多个第一图像及其对应的第一标注信息，其中，所述第一图像包括至少一个第一目标，所述第一标注信息包括所述第一目标对应的第一边界框和类别；

基于所述第一图像集，训练数据处理模型；

获取第二图像集，所述第二图像集包括多个第二图像及其对应的第二标注信息，其中，所述第二图像包括至少一个第二目标，所述第二标注信息包括所述第二目标对应的第二标注点和类别；

基于所述第二图像集，通过完成训练的所述数据处理模型，得到第三图像集，所述第三图像集包括所述多个第二图像及其对应的第三标注信息，其中，所述第三标注信息包括所述第二目标对应的第二边界框和类别。

2.根据权利要求1所述的图像数据处理方法，其中，所述基于所述第一图像集，训练数据处理模型，包括：

在所述第一图像的每个第一边界框内生成对应所述第一目标的第一标注点；

基于所述第一图像、以及所述第一图像中所述第一目标对应的第一标注点及类别，通过待训练的数据处理模型得到对应于所述第一目标的第一预测框；

基于对应于同一第一目标的所述第一预测框与所述第一边界框得到损失值；

基于所述损失值调整待训练的数据处理模型的参数，得到完成训练的所述数据处理模型。

3.根据权利要求2所述的图像数据处理方法，其中，所述基于所述第一图像集，训练数据处理模型，还包括：

对所述第一标注点的坐标进行归一化。

4.根据权利要求2或3所述的图像数据处理方法，其中，所述损失值包括以下一种或多种，L1损失、IOU损失及GIOU损失。

5.根据权利要求2或3所述的图像数据处理方法，其中，所述基于所述第二图像集，通过完成训练的所述数据处理模型，得到第三图像集，包括：

基于所述第二目标对应的第二标注点和类别，通过完成训练的所述数据处理模型，得到所述第二目标对应的第二边界框。

6.一种目标检测模型训练方法，其中，所述目标检测模型用于图像目标检测，所述目标检测模型训练方法包括：

获取训练集，所述训练集至少包括采用如权利要求1-5所述的图像数据处理方法得到的所述第三图像集；

基于所述训练集，训练待训练的目标检测模型，得到训练完成的所述目标检测模型。

7.根据权利要求6所述的目标检测模型训练方法，其中，所述训练集还包括所述第一图像集。

8.根据权利要求7所述的目标检测模型训练方法，其中，所述基于所述训练集，训练待训练的目标检测模型，得到训练完成的所述目标检测模型，包括：

基于所述第一图像或第二图像，通过待训练的目标检测模型，得到所述第一目标或第二目标的预测框；

基于对应于同一第一图像的预测框和第一边界框，或对应于同一第二图像的预测框和第二边界框，得到边界框损失值；

基于所述边界框损失值调整待训练的目标检测模型的参数，得到完成训练的目标检测模型。

9.一种目标检测方法，其中，所述目标检测方法包括：

获取图像，所述图像包括至少一个待检测目标；

通过目标检测模型，确定所述待检测目标的边界框，其中，所述目标检测模型通过如权利要求6-8任一项所述的目标检测模型训练方法训练得到。