[发明专利]基于相关性图学习的图像识别方法及装置

说明书

本发明是关于一种基于相关性图学习的图像识别方法和装置，其中，方法包括：获取待识别图像；使用训练好的特征提取网络确定所述待识别图像对应的特征向量；从训练集中随机获取每个类别的图集的多个聚类中心；将每个类别的图集的多个聚类中心与所述待识别图像的特征向量构成图；使用图卷积神经网络对所述图进行类别预测，以确定所述待识别图像所属的图像类别。通过该技术方案，充分模拟了人类的经验系统，提高了图像识别的效果。

技术领域

本发明涉及图像识别技术领域，尤其涉及一种基于相关性图学习的图像识别方法和装置。

背景技术

图像识别以图像的主要特征为基础，每个图像都有明显或不明显的特征，例如数字1由一条竖线、数字0有一个圆，而数字8有两个圆，我们之所以能够快速识别出来，是因为我们第一眼就看到了它们的这些主要特征。由此可见，在图像识别过程中，知觉机制必须排除输入的多余信息,抽出关键的信息。同时,在大脑里必定有一个负责整合信息的机制，它能把分阶段获得的信息整理成一个完整的知觉映象。

现有的图像识别方法主要包括基于传统机器学习方法和基于深度学习方法。传统机器学习方法在分类性能上提升的空间有限，准确率不高；深度学习在计算机视领域迅速发展，在许多图像识别任务中，击败了很多的传统机器学习方法，基于深度学习的方法具有以下优点：（1）学习能力强，具有较高的识别性能，准确率高。（2）深度学习的神经网络层数很多，宽度很广，理论上可以映射到任意函数，所以能解决很复杂的问题。基于以上优点，深度学习方法已成为一种主流的方法。

深度学习方法已成为图像识别中一种主流的方法。而在图像识别方面，人类之所以能够快速识别，是因为他们经验丰富，看过很多幅图像。人们可以运用以前积累起来的经验，在对一幅图像识别时，可以使用以前看过的并且比较典型的图像来辅助完成对当前图像的识别。而现在的深度学习方法几乎没有模仿利用人类的经验信息来辅助识别，对样本之间的关联性考虑不多，在对一幅图像识别时，很难将其他图像的关联性考虑进去，不能很好地使用其他图像来辅助当前图像的识别，从而影响了当前图像的识别效果。

在已有的识别方法中，基于相关性图学习的模型的方法具有以下优点：（1）对于噪声具有较好的鲁棒性。（2）模型参数量，训练速度快。与传统的识别方法不同，该类模型参数量很少，训练速度快，便于部署并且可以快速实现。（3）从建模的角度来说，该类方法更容易加入先验知识。基于以上优点，相关性图学习已成为一种主流的方法。近年来，大家都是用深度学习的方法来完成对图像的识别，但是现有深度学习的方法都是对每一张图像单独进行识别，并没有考虑不同图像之间的相似性与差异性，没有使用其他图像来辅助完成当前图像的识别，没有模拟人类的经验系统，所以识别效果较差。因此，如何使用其他图像来辅助完成当前图像的识别，提高图像识别的效果，是目前本领域技术人员需要解决的技术问题。

发明内容

为克服相关技术中存在的问题，本发明提供一种基于相关性图学习的图像识别方法和装置，使用相关性图学习的方法通过其他图像来辅助完成当前图像的识别分类，更容易加入先验知识，适当考虑了不同图像之间的相似性与差异性，充分模拟了人类的经验系统，提高了图像识别的效果。

根据本发明实施例的第一方面，提供一种基于相关性图学习的图像识别方法，所述方法包括：

获取待识别图像；

使用训练好的特征提取网络确定所述待识别图像对应的特征向量；

从训练集中随机获取每个类别的图集的多个聚类中心；

将每个类别的图集的多个聚类中心与所述待识别图像的特征向量构成图；

使用图卷积神经网络对所述图进行类别预测，以确定所述待识别图像所属的图像类别。

在一个实施例中，优选地，在获取待识别图像之前，所述方法还包括：

构造图像识别数据集，其中，所述图像识别数据集包括训练集和测试集；

使用训练集中的图像训练得到所述特征提取网络；