[发明专利]训练优化的基于卷积神经网络的色情图像或视频检测方法

说明书

技术领域

本发明涉及一种色情图像检测方法，更具体地说，涉及一种训练优化的基于卷积神经网络的色情图像或视频检测方法。

背景技术

随着互联网的广泛应用，用户在获取大量有用信息的同时也会遇到不良信息，尤其以色情图像和色情视频最严重。现有的色情图像或色情视频检测的方法主要是利用肤色进行识别，但是在实际应用中，现有的检测方法存在以下限制检测准确率的因素：

1)环境光线的多样化、人种的多样性等使得很难建立完善的肤色模型及检测的准确率；

2)人体姿态多样性、遮挡等外界因素使得检测的准确率一直不高；

3)对于不同的色情视频难以建立比较通用的规则。

发明内容

本发明的目的在于克服现有技术的不足，提供一种不仅可以适应各种环境和姿态以及遮挡等外界因素的影响，而且能够建立比较通用的检测规则，提高检测的准确率的训练优化的基于卷积神经网络的色情图像或视频检测方法。

本发明的技术方案如下：

一种训练优化的基于卷积神经网络的色情图像或视频检测方法，预设图像数据或视频数据；并且利用已进行卷积神经网络训练得到的初步检测模型进行第一次分类检测；如果第一次分类检测结果存在错误的样本，对错误的样本，利用卷积神经网络进行第二次训练，得到第一次过渡检测模型；并利用第一次过渡检测模型进行第二次分类检测，如果第二次分类检测结果存在错误的样本，对错误的样本，利用卷积神经网络进行第三次训练，得到第二次过渡检测模型，循环以上步骤，直至检测结果中错误的样本少于预设的阀值或者不存在，得到最终检测模型；利用最终检测模型对待测的图像数据或视频数据进行分类检测。

作为优选，预设图像数据或视频数据包括预分类的色情数据和正常数据，将这两类数据作为两个场景，作为输入数据提供给卷积神经网络进行学习分类，得到判断图像数据或视频数据是否为色情和正常的检测模型。

作为优选，训练优化的过程为：将得到的过渡检测模型对预设的图像数据或视频数据进行分类测试，将分类检测结果与预分类的结果不同的图像数据或视频数据定为分类错误的样本。

作为优选，分类错误的样本具体为：将分类检测结果获取的环境分类标签与预分类的环境分类标签不一致的样本图像数据或视频数据进行收集，重新进行分类。

作为优选，卷积神经网络的网络结构的顺序为：输入层、若干小组层、全连接层、概率统计层。

作为优选，进一步地，小组层依次包括卷积层、激活层。

作为优选，卷积层中，核的大小必须是奇数，且不大于该层输入的宽或者高；中间表示通过卷积层时不改变宽和高，通道数不限制是否改变；激活层中，激活层不改变卷积层表示的宽、高、通道数；激活层设置在卷积层或者全连接层之后；通过全连接层后的中间表示为1维，而且经过全连接层后，无法再经过小组层进行处理；概率统计层将全连接层产生的实值变成[0,1]之间的概率。

作为优选，概率统计层输出每一种标签的概率，并选择概率最大的标签作为当前图像的类型。

作为优选，进一步地，激活层后设置下采样层与归一化层的一层或两层。

作为优选，下采样层不改变中间表示的通道数，而且下采样层对图像的缩小比为核的大小；归一化层不改变中间表示的任何尺寸。

本发明的有益效果如下：

本发明所述的方法，经过循环训练，得到最终的检测模型，能够最大程序地提高分类检测准确率，克服因环境光线的多样化、人种的多样性等使得很难建立完善的肤色模型及检测的准确率，以及人体姿态多样性、遮挡等外界因素使得检测的准确率一直不高等不足。

本发明所述的方法用于图像数据或视频数据时，进行对图片的检测，或者对视频图像序列的检测，只要分类结果为色情，即可定义该图像集合或视频为色情数据，克服对于不同的色情视频难以建立比较通用的规则的不足。

具体实施方式

以下结合实施例对本发明进行进一步的详细说明。