[发明专利]敏感图像识别方法及装置

说明书

本发明公开了一种敏感图像识别方法及装置，属于图像识别领域。所述方法包括：获取待识别图像；检测敏感图像数据库中是否存在目标图像，得到检测结果，敏感图像数据库用于存储多个敏感图像以及每个敏感图像的所属类别，目标图像为与待识别图像相同的图像；通过图像识别模型对待识别图像进行类别预测，得到预测结果，预测结果包括待识别图像的多个可能所属类别以及每个可能所属类别的概率；根据检测结果和预测结果，得到待识别图像的所属类别。通过结合基于敏感图像数据库的识别方法和基于图像识别模型的识别方法，实现在识别典型敏感图像的同时，能够识别新出现的或传播范围小的敏感图像，扩展了敏感图像的识别范围，识别效果好。

技术领域

本发明涉及图像识别领域，特别涉及一种敏感图像识别方法及装置。

背景技术

随着互联网的普及，互联网已经成为各种信息传播的重要途径。用户在通过互联网获取大量有用信息的同时，还会获取到大量不良信息，比如内容涉及色情、招嫖、暴力、政治敏感等法律法规禁止传播的敏感图像等。这些不良信息会对用户造成不良影响，因此，需要对敏感图像进行识别，进而净化网络环境。

现有技术中，对敏感图像的识别过程为：建立敏感图像数据库，通常情况下，该敏感图像数据库中包含属于相同敏感图像类别的图像指纹；在获取到待识别图像时，提取待识别图像的图像指纹；之后，若在图像数据库中获取到与待识别图像的图像指纹匹配的目标图像指纹，则确定待识别图像的类别为该敏感图像类别。

在实现本发明的过程中，发明人发现现有技术至少存在以下问题：

上述敏感图像识别方法依赖于敏感图像数据库，敏感图像数据库的建立受到各种条件的限制，比如存储资源限制、查找速度限制、人力限制等，其中敏感图像数据库的存储资源和查找速度的限制不允许太多的敏感图像添加。并且，人力限制会导致不能及时地将新出现的不同类别的敏感图像进行审核，进而不能及时对敏感图像数据库进行更新，使得基于敏感数据库仅识别出实际传播敏感图像中的极少比例的敏感图像。因此，上述敏感图像识别方法的识别范围窄，识别效果差。

发明内容

为了解决现有技术的问题，本发明实施例提供了一种敏感图像识别方法及装置。所述技术方案如下：

一方面，提供了一种敏感图像识别方法，所述方法包括：

获取待识别图像；

检测敏感图像数据库中是否存在目标图像，得到检测结果，所述敏感图像数据库用于存储多个敏感图像以及每个敏感图像的所属类别，所述目标图像为与所述待识别图像相同的图像；

通过图像识别模型对所述待识别图像进行类别预测，得到预测结果，所述预测结果包括所述待识别图像的多个可能所属类别以及每个可能所属类别的概率，所述图像识别模型根据敏感图像样本进行学习建立；

根据所述检测结果和所述预测结果，得到所述待识别图像的所属类别。

另一方面，提供了一种敏感图像识别装置，所述装置包括：

获取模块，用于获取待识别图像；

检测模块，用于检测敏感图像数据库中是否存在目标图像，得到检测结果，所述敏感图像数据库用于存储多个敏感图像以及每个敏感图像的所属类别，所述目标图像为与所述待识别图像相同的图像；

类别预测模块，用于通过图像识别模型对所述待识别图像进行类别预测，得到预测结果，所述预测结果包括所述待识别图像的多个可能所属类别以及每个可能所属类别的概率，所述图像识别模型根据敏感图像样本进行学习建立；

类别确定模块，用于根据所述检测结果和所述预测结果，得到所述待识别图像的所属类别。

本发明实施例提供的技术方案带来的有益效果是：