[发明专利]一种检测模型的训练方法及装置

权利要求书

1.一种检测模型的训练方法，包括：

从样本集合中随机选择具有第一预定数量样本的第一未标注样本集；

对所述第一未标注样本集中的每一样本进行标注，得到第一已标注样本集；

对所述第一已标注样本集中的每一样本进行数据增强，获得第一增强样本集；

通过所述第一增强样本集对待训练的检测模型进行第一次训练；

从所述样本集合中随机选择具有第二预定数量样本的第二未标注样本集；

利用所述经过第一次训练后的检测模型对所述第二未标注样本集中的每一样本进行检测，得到第一检测结果集；

对所述第一检测结果集中的每一检测结果进行数据增强，得到第二增强样本集；

通过所述第二增强样本集对所述第一次训练后的检测模型进行第二次训练，得到训练后的检测模型。

2.如权利要求1所述的方法，还包括：

判断第一次训练后的检测模型是否满足预设条件；

当第一次训练后的检测模型满足预设条件时，执行从所述样本集合中随机选择具有第二预定数量样本的第二未标注样本集的步骤。

3.如权利要求2所述的方法，其中，所述判断第一次训练后的检测模型是否满足预设条件，包括：

获取第二已标注样本集；

利用第一次训练后的检测模型对所述第二已标注样本集进行检测，得到第二检测结果；

根据所述第二检测结果与所述第二已标注样本集中的标注信息，确定第一次训练后的检测模型的检测准确度；

判断第一次训练后的检测模型的检测准确度是否满足预设条件。

4.如权利要求1所述的方法，其中，所述对所述第一已标注样本集中的每一样本进行数据增强，包括：

从预设的数据增强方法集合中随机选取一种或多种数据增强方法；

利用所述随机选取出的一种或多种数据增强方法对所述第一已标注样本集中的每一样本进行数据增强；

所述对所述第一检测结果集中的每一检测结果进行数据增强，包括：

从预设的数据增强方法集合中随机选取一种或多种数据增强方法；

利用选取出的一种或多种数据增强方法对所述第一检测结果集中的每一检测结果进行数据增强。

5.如权利要求4所述的方法，所述数据增强方法，包括：

在预设尺度范围内随机放缩数据、在预设角度范围内随机旋转数据、以及在数据中随机加入高斯噪声。

6.如权利要求4所述的方法，其中，在所述利用所述随机选取出的一种或多种数据增强方法对所述第一已标注样本集中的每一样本进行数据增强的步骤之前，所述方法还包括：

确定所述随机选取的一种或多种数据增强方法中的每一种数据增强方法的参数。

7.如权利要求6所述的方法，其中，所述确定所述随机选取的一种或多种数据增强方法中的每一种数据增强方法的参数，包括：

获取每一种数据增强方法各自对应的预设参数选取范围；

在所述预设参数选取范围中随机选择参数。

8.一种检测模型的训练装置，包括：

第一选择模块，用于从样本集合中随机选择具有第一预定数量样本的第一未标注样本集；

第一标注模块，用于对所述第一未标注样本集中的每一样本进行标注，得到第一已标注样本集；

第一增强模块，用于对所述第一已标注样本集中的每一样本进行数据增强，获得第一增强样本集；

第一训练模块，用于通过所述第一增强样本集对待训练的检测模型进行第一次训练；

第二选择模块，用于从所述样本集合中随机选择具有第二预定数量样本的第二未标注样本集；

第一检测模块，用于利用所述经过第一次训练后的检测模型对所述第二未标注样本集中的每一样本进行检测，得到第一检测结果集；

第二增强模块，用于对所述第一检测结果集中的每一检测结果进行数据增强，得到第二增强样本集；

第二训练模块，用于通过所述第二增强样本集对所述第一次训练后的检测模型进行第二次训练，得到训练后的检测模型。