[发明专利]人头检测模型生成方法、装置、人头检测模型及人头检测方法

权利要求书

1.一种人头检测模型的生成方法，其特征在于，包括：

获取初始模型和训练样本集，所述训练样本集包括样本图像，所述样本图像中包含头部正例训练样本、头部标注信息、手部负例训练样本以及手部标注信息；

根据样本图像得到第一训练图像与第二训练图像，其中，第一训练图像包含头部正例训练样本与头部标注信息，第二训练图像包含手部负例训练样本；

根据所述第一训练图像以及第二训练图像对初始模型进行训练，得到人头检测模型。

2.根据权利要求1所述的一种人头检测模型的生成方法，其特征在于，还包括：

对所述第一训练图像以及第二训练图像进行数据增强，并生成训练图片；所述训练图片中包含头部正例训练样本、手部负例训练样本以及头部标注信息；

根据所述训练图片以及头部标注信息对初始模型进行训练得到人头检测模型。

3.根据权利要求1所述的一种人头检测模型的生成方法，其特征在于，所述第一训练图像或第二训练图像的获取方法包括：

对样本图像中的头部正例训练样本或手部负例训练样本进行标注生成样本图像标注框，对样本图像标注框对应的样本图像进行抠图；

所述抠图范围满足如下公式：

x_min≥0；

x_max≤width；

y_min≥0；

y_max≤height；

其中，x_label-min为样本图像标注框的x轴的最小值，x_label-max为样本图像标注框的x轴的最大值，y_label-min为样本图像标注框的y轴的最小值，y_label-max为样本图像标注框的y轴的最大值，width为样本图像的宽，height为样本图像的高，a为自定义系数。

4.根据权利要求1或3所述的一种人头检测模型的生成方法，其特征在于，还包括：所述检测头部目标的位置信息的获取方法包括：

x_new-hand-max＝x_new-hand-min+(x_label-max-x_label-min)；

y_new-hand-max＝y_new-hand-min+(y_label-max-y_label-min)；

其中，x和y分别为训练图片中第一训练图像或第二训练图像放置的左上角坐标，width为样本图像的宽，height为样本图像的高。

5.根据权利要求1所述的一种人头检测模型的生成方法，其特征在于，还包括，所述训练图片中第一训练图像或第二训练图像的左上角坐标的获取方法包括：

其中，width_m为第一训练图像或第二训练图像的宽，height_m为第一训练图像或第二训练图像的高，num为第一训练图像或第二训练图像的放置序号。

6.根据权利要求5所述的一种人头检测模型的生成方法，其特征在于，还包括：

获取训练图片中的第一训练图像和/或第二训练图像的左上角坐标；

对所述第一训练图像和/或第二训练图像的左上角坐标进行调节，得到最终训练图片。

7.根据权利要求6所述的一种人头检测模型的生成方法，其特征在于，所述第一训练图像和/或第二训练图像的左上角坐标的调节范围满足如下公式：

其中，β和γ为函数自动生成的随机整数。