[发明专利]用于色情图片的图像数据处理方法及装置

权利要求书

1.一种用于色情图片的图像数据处理方法，其特征在于，包括：

生成训练数据集，其中，训练数据集包括：预设人体关键点标签；

根据所述训练数据集训练得到预训练模型；

输入待检测图片；以及

根据预训练模型判断所述待检测图片是否属于色情图片。

2.根据权利要求1所述的图像数据处理方法，其特征在于，生成训练数据集包括：

提取包含有相关要素的图片中的人体关键点信息；

根据所述人体关键点信息对人体执行裁剪操作；以及

将裁剪得到的图片和对应的已标注有相关要素的标签储存到训练集数据库。

3.根据权利要求1或2所述的图像数据处理方法，其特征在于，生成训练数据集包括：

根据关键点对人体执行裁剪；以及

将人体图像中的肩部位置的关键点在原图中的中央位置对齐。

4.根据权利要求1所述的图像数据处理方法，其特征在于，根据预训练模型判断所述待检测图片是否属于色情图片包括：

在预训练模型中输入与训练阶段相同分布的随机变量及拟得到的图片标签；以及

根据预训练模型中生成的对抗网络，获取生成的图片。

5.根据权利要求1所述的图像数据处理方法，其特征在于，根据预训练模型判断所述待检测图片是否属于色情图片还包括：模型训练过程，所述模型训练过程包括：

在生成网络中生成类似真实的预设目标图片；

在判断网络中判断一个给定预设图片是不是真实的预设目标图片；以及

将上述生成网络和判断网络持续对抗训练，并通过预测的判决结果与真实的结果之间的差别，更新生成网络和判决网络中的参数。

6.一种用于色情图片的图像数据处理装置，其特征在于，包括：

生成模块，用于生成训练数据集，其中，训练数据集包括：预设人体关键点标签；

训练模块，用于根据所述训练数据集训练得到预训练模型；

输入模块，用于输入待检测图片；以及

判断模块，用于根据预训练模型判断所述待检测图片是否属于色情图片。

7.根据权利要求6所述的图像数据处理装置，其特征在于，所述生成模块包括：

提取单元，用于提取包含有相关要素的图片中的人体关键点信息；

裁剪单元，用于根据所述人体关键点信息对人体执行裁剪操作；以及

存储单元，用于将裁剪得到的图片和对应的已标注有相关要素的标签储存到训练集数据库。

8.根据权利要求6所述的图像数据处理装置，其特征在于，所述生成模块包括：

关键点裁剪单元，用于根据关键点对人体执行裁剪；以及

对齐单元，用于将人体图像中的肩部位置的关键点在原图中的中央位置对齐。

9.根据权利要求6所述的图像数据处理装置，其特征在于，所述判断模块包括：

变量和标签输入单元，用于在预训练模型中输入与训练阶段相同分布的随机变量及拟得到的图片标签；以及

判决单元，用于根据预训练模型中生成的对抗网络，获取生成的图片。

10.根据权利要求6所述的图像数据处理装置，其特征在于，还包括：模型训练过程模块，所述模型训练过程模块包括：

生成网络单元，用于在生成网络中生成类似真实的预设目标图片；

判断网络单元，用于在判断网络中判断一个给定预设图片是不是真实的预设目标图片；以及

更新单元，用于将上述生成网络和判断网络持续对抗训练，并通过预测的判决结果与真实的结果之间的差别，更新生成网络和判决网络中的参数。