[发明专利]图像分类的方法和装置

说明书

本公开是关于一种图像分类的方法和装置，通过获取图像的主题频次和类别矩阵，根据图像的主题频次和类别矩阵，利用分类器，获取图像的类别，考虑了整幅图像的所有主题的频次，使得分类的参考因素更加丰富，提高了图像分类的准确性，并且，避免了大量人工标记的工作，提高了图像分类的效率。

技术领域

本公开涉及图像识别技术，尤其涉及一种图像分类的方法和装置。

背景技术

随着计算机技术的飞速发展，人们获取图像的方式越来越便利，存储的图像数量也越来越多，对于大量的图像进行分类管理成为一种需求，图像分类是将图像标记为不同类别的过程，图像类别如城市、森林、沙滩、动物和植物等。

相关技术中，通过每幅图像的局部纹理特征确定一幅图像的类别，然而，图像分类的准确性不高。

发明内容

为克服相关技术中存在的问题，本公开提供一种图像分类的方法和装置。

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

获取图像的主题频次；

获取类别矩阵，所述类别矩阵中包含C个类别的向量，每个类别的向量服从正态分布，所述C为大于等于2的整数；

根据所述图像的主题频次和所述类别矩阵，利用分类器，获取所述图像的类别。

可选地，所述获取图像的主题频次，包括：

以预设间距提取图像的所有相邻视觉词汇对，所述相邻视觉词汇对中包含第一视觉词汇和第二视觉词汇，所述第一视觉词汇对应的提取位置和所述第二视觉词汇对应的提取位置相邻；

获取每个相邻视觉词汇对对应的主题对；

根据所述图像的所有相邻视觉词汇对对应的主题对，确定所述图像的主题频次。

可选地，所述获取每个相邻视觉词汇对对应的主题对，包括：

获取所述图像对应的相邻主题概率矩阵；

获取每个主题对应的视觉词汇概率向量；

根据所述相邻主题概率矩阵和所述每个主题对应的视觉词汇概率向量，获取每个相邻视觉词汇对对应的主题对。

可选地，所述根据所述相邻主题概率矩阵和所述每个主题对应的视觉词汇概率向量，获取每个相邻视觉词汇对对应的主题对，包括：

获取每个相邻视觉词汇对的N个候选主题对，其中，所述候选主题对包含第一候选主题和第二候选主题；

针对所述N个候选主题对中的每个所述候选主题对，根据所述相邻主题概率矩阵和所述每个主题对应的视觉词汇概率向量，获取所述相邻视觉词汇对为所述候选主题对的概率；

根据获取的N个所述相邻视觉词汇对为所述候选主题对的概率，确定最大概率对应的候选主题对为所述相邻视觉词汇对对应的主题对。

可选地，所述根据所述相邻主题概率矩阵和所述每个主题对应的视觉词汇概率向量，获取所述相邻视觉词汇对为所述候选主题对的概率，包括：

根据所述相邻主题概率矩阵，获取所述候选主题对为相邻主题的第一概率；

根据所述每个主题对应的视觉词汇概率向量，获取所述第一候选主题为第一视觉词汇的第二概率，以及所述第二候选主题为所述第二视觉词汇的第三概率；

根据所述第一概率、所述第二概率和所述第三概率，获取所述相邻视觉词汇对为所述候选主题对的概率。

可选地，所述获取所述图像对应的相邻主题概率矩阵，包括：