[发明专利]图像分类方法、装置、电子设备、计算机可读介质

说明书

本公开提供了图像分类方法、装置、电子设备、计算机可读介质，涉及图像处理技术领域，尤其涉及人工智能、计算机视觉等技术领域。具体实现方案为：将待分类图像输入预先训练的目标检测网络，获取所述待分类图像中的至少一个候选区域；将所述候选区域对应的候选区域图像输入预先训练的图像特征提取网络获取所述候选区域对应的图像特征；将所述待分类图像输入预先训练的文字识别网络，获取所述待分类图像中的文字对应的文字特征；将所述图像特征和所述文字特征进行拼接获取拼接特征，并将所述拼接特征输入预先训练的分类网络，获取所述待分类图像对应的分类结果。

技术领域

本公开涉及图像处理技术领域，尤其涉及人工智能、计算机视觉等技术领域。具体而言，本公开涉及一种图像分类方法、装置、电子设备、计算机可读介质。

背景技术

一些特殊形式的图像(如漫画等)由于夸张等艺术特点，往往具有危害性，且这些图像往往具有极强的隐蔽性，不易被发现。

如在社交平台、直播平台都不允许有害漫画的传播，但由于漫画具有极强的隐蔽性，增加了社交平台、直播平台的风控难度。

发明内容

本公开提供了一种图像分类方法、装置、电子设备、计算机可读介质。

根据本公开的第一方面，提供了一种图像分类的方法，该方法包括：

将待分类图像输入预先训练的目标检测网络，获取所述待分类图像中的至少一个候选区域；

将所述候选区域对应的候选区域图像输入预先训练的图像特征提取网络获取所述候选区域对应的图像特征；

将所述待分类图像输入预先训练的文字识别网络，获取所述待分类图像中的文字对应的文字特征；

将所述图像特征和所述文字特征进行拼接获取拼接特征，并将所述拼接特征输入预先训练的分类网络，获取所述待分类图像对应的分类结果。

根据本公开的第二方面，提供了一种图像分类的装置，包括：

区域模块，用于将待分类图像输入预先训练的目标检测网络，获取所述待分类图像中的至少一个候选区域；

图像模块，用于将所述候选区域对应的候选区域图像输入预先训练的图像特征提取网络获取所述候选区域对应的图像特征；

文字模块，用于将所述待分类图像输入预先训练的文字识别网络，获取所述待分类图像中的文字对应的文字特征；

分类模块，用于将所述图像特征和所述文字特征进行拼接获取拼接特征，并将所述拼接特征输入预先训练的分类网络，获取所述待分类图像对应的分类结果。

根据本公开的第三方面，提供了一种电子设备，该电子设备包括：

至少一个处理器；以及

与上述至少一个处理器通信连接的存储器；其中，

存储器存储有可被上述至少一个处理器执行的指令，指令被上述至少一个处理器执行，以使上述至少一个处理器能够执行上述图像分类方法。

根据本公开的第四方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，其中，该计算机指令用于使计算机执行上述图像分类方法。

根据本公开的第五方面，提供了一种计算机程序产品，包括计算机程序，该计算机程序在被处理器执行时实现上述图像分类方法。

应当理解，本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征，也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。

附图说明

附图用于更好地理解本方案，不构成对本公开的限定。其中：