[发明专利]图片预标注方法、装置及终端设备

权利要求书

1.一种图片预标注方法，其特征在于，包括：

采用第一数量的第一图片训练轻量级网络，得到第一模型，其中，所述第一图片中的目标物体的类别和位置均被人工通过目标框预标注；

采用所述第一模型对第二数量的第二图片进行预标注，得到包含第一预标注信息的第二图片，若所述第一预标注信息的置信度大于或等于预设置信度阈值，则将置信度大于或等于预设置信度阈值对应的包含第一预标注信息的第二图片存入预标注数据集；

采用所述第一图片以及所述预标注数据集中的包含第一预标注信息的第二图片训练重量级网络，得到第二模型；

采用所述第二模型对剩余需要预标注的图片进行预标注。

2.如权利要求1所述的图片预标注方法，其特征在于，所述第一模型的个数大于或等于2，且各个第一模型为不同的模型，所述采用所述第一模型对第二数量的第二图片进行预标注，得到包含第一预标注信息的第二图片，若所述第一预标注信息的置信度大于或等于预设置信度阈值，则将置信度大于或等于预设置信度阈值对应的包含第一预标注信息的第二图片存入预标注数据集，包括：

采用各个所述第一模型对第二数量的第二图片进行预标注，得到包含各个第一预标注信息的第二图片，若各个第一预标注信息指示各个所述第一模型均能在同一第二图片中检测出相同的目标物体，且所述各个第一预标注信息的置信度均大于或等于预设置信度阈值，则将置信度大于或等于预设置信度阈值对应的包含第一预标注信息的第二图片存入预标注数据集。

3.如权利要求2所述的图片预标注方法，其特征在于，在所述得到第二模型之后，包括：

分别采用所述第二模型和各个所述第一模型对第三数量的第三图片进行预标注，得到包含第二预标注信息的第三图片，所述第二预标注信息用于通过目标框指示在所述第三图片识别出的目标物体的类别和位置；

若所述第二预标注信息指示所述第二模型和各个所述第一模型中，识别出所述第三图片中的同一目标物体的模型的个数超过50％，则分别统计所述第二模型准确识别出的目标框的数量和各个所述第一模型准确识别出的目标框的数量；

所述采用所述第二模型对剩余需要预标注的图片进行预标注包括：

若所述第二模型准确识别出的目标框的数量大于各个所述第一模型准确识别出的目标框的数量，则采用所述第二模型对剩余需要预标注的图片进行预标注。

4.如权利要求3所述的图片预标注方法，其特征在于，在所述分别采用所述第二模型和各个所述第一模型对第三数量的第三图片进行预标注之前，包括：

在对所述预标注数据集中包含第一预标注信息的第二图片进行校验后，采用校验后的包含第一预标注信息的第二图片继续训练所述第一模型，得到训练后的所述第一模型。

5.如权利要求3所述的图片预标注方法，其特征在于，所述采用所述第二模型对剩余需要预标注的图片进行预标注，包括：

采用所述第二模型对第四数量的第四图片进行预标注，得到包含第三预标注信息的第四图片，并将所述包含第三预标注信息的第四图片存入预标注数据集；

在对所述预标注数据集中的所述包含第三预标注信息的第四图片进行校验后，采用所述预标注数据集中的各个图片继续训练所述第二模型，其中，第四数量的第四图片为剩余需要预标注的部分图片；

采用训练后的所述第二模型对除所述第四数量的第四图片后的剩余需要预标注的图片进行预标注。

6.如权利要求5所述的图片预标注方法，其特征在于，在所述采用所述第二模型对第四数量的第四图片进行预标注，得到包含第三预标注信息的第四图片，并将所述包含第三预标注信息的第四图片存入预标注数据集之前，还包括：

若所述第二预标注信息指示所述第二模型和各个所述第一模型中，识别出所述第三图片中的同一目标物体的模型的个数超过50％，且识别所述同一目标物体对应的置信度均大于或所述等于预设置信度阈值，则将所述同一目标物体的目标框存入所述预标注数据集；

若所述第二预标注信息指示所述第二模型和各个所述第一模型中，识别出所述第三图片中的同一目标物体的模型的个数超过50％，且识别所述同一目标物体对应的置信度不全部大于或等于所述预设置信度阈值，则在对所述同一目标物体的目标框进行校验后再存入所述预标注数据集。