[发明专利]一种图像分类方法及卷积神经网络生成方法

说明书

本发明公开了一种图像分类方法、用于对图像进行分类处理的卷积神经网络生成方法、用于对图像中的文字进行识别的卷积神经网络生成方法、移动终端及计算设备，所述图像分类方法适于在移动终端中执行，所述移动终端包括图像库，所述图像库中存储有多张图像，所述图像分类方法包括步骤：对图像库中的每一张图像，对该图像进行分类处理以获取其对应的类别；若所述类别为文本类，则对该图像进行文字识别，以提取出该图像所包含的文本信息；将所述文本信息与该图像的图像存储路径及图像名称进行关联存储。

技术领域

本发明涉及图像处理技术领域，特别涉及一种图像分类方法、用于对图像进行分类处理的卷积神经网络生成方法、用于对图像中的文字进行识别的卷积神经网络生成方法、移动终端及计算设备。

背景技术

随着硬件技术的不断发展，越来越多的人开始使用如智能手机、平板电脑等移动终端进行照片拍摄并存储，以记录下珍贵的瞬间。当移动终端中所保存的照片数量越来越多时，由于照片繁多，又都分属不同的类别，常常会出现无法及时寻找出某一张照片的情况，给用户带来较差的体验。

现有的图像分类算法，通常是将移动终端的相册里的图像分为各个类别，按照类别进行图像管理，但不会再进行进一步操作。这样的处理方式虽然方便用于根据类别来查找图像，但如果要快速定位到某一张具有特定信息的图像则无法实现，比如说用户对一张图像上面的部分文字有印象，但其他内容却不记得，此时就难以通过用户自身对该图像的类别认知，来迅速且准确地获取到所需图像及图像中所包含的信息。因此，需要提供一种新的图像分类方法来改进上述处理过程。

发明内容

为此，本发明提供一种图像分类方案，并提供了用于对图像进行分类处理的卷积神经网络生成方案和用于对图像中的文字进行识别的卷积神经网络生成方案，以力图解决或者至少缓解上面存在的问题。

根据本发明的一个方面，提供一种图像分类方法，适于在移动终端中执行，移动终端包括图像库，图像库中存储有多张图像，该方法包括如下步骤：首先，对图像库中的每一张图像，对该图像进行分类处理以获取其对应的类别；若类别为文本类，则对该图像进行文字识别，以提取出该图像所包含的文本信息；将文本信息与该图像的图像存储路径及图像名称进行关联存储。

可选地，在根据本发明的图像分类方法中，当接收到用户键入的检索词时，该方法还包括：根据检索词查找是否存在与其相同或相似的文本信息；若存在，则获取该文本信息关联的图像存储路径；根据该图像存储路径查找到其对应的图像，将该图像和该文本信息向用户展示。

可选地，在根据本发明的图像分类方法中，对该图像进行文字识别，以提取出该图像所包含的文本信息的步骤包括：获取该图像所包含的各单个文字对应的文字图像区域；分别对各文字图像区域进行文字识别，以确定各文字图像区域所包含的文字；基于各文字生成该图像对应的文本信息。

可选地，在根据本发明的图像分类方法中，基于各文字生成该图像对应的文本信息的步骤包括：获取该图像中各文字图像区域之间的位置关系；根据位置关系，对各文字图像区域对应的文字进行组合，以生成该图像对应的文本信息。

可选地，在根据本发明的图像分类方法中，移动终端中存储有用于对图像进行分类处理的、训练好的第一卷积神经网络，对该图像进行分类处理以获取其对应的图像类型的步骤包括：将该图像输入到训练好的第一卷积神经网络中进行图像分类；根据该第一卷积神经网络的输出确定该图像的类别。

可选地，在根据本发明的图像分类方法中，移动终端中存储有用于对图像中的文字进行识别的、训练好的第二卷积神经网络，对该图像进行文字识别，以提取出该图像所包含的文本信息的步骤包括：获取该图像所包含的各单个文字对应的文字图像区域；分别将各文字图像区域输入到训练好的第二卷积神经网络中进行文字识别，根据该第二卷积神经网络的输出确定各文字图像区域所包含的文字；基于各文字生成该图像对应的文本信息。