[发明专利]图片分类方法、装置、电子设备

说明书

本申请公开了一种图片分类方法，属于计算机技术领域，有助于对上传至网络平台的图片进行准确分类。本申请实施例公开的图片分类方法包括：获取目标图片，以及所述目标图片的标签文本和描述文本；基于空间注意力机制确定所述目标图片的图片内容的向量表示，以及，分别确定所述标签文本和所述描述文本的向量表示；基于所述图片内容的向量表示、所述标签文本和所述描述文本的所述向量表示，对所述目标图片进行分类处理。本方法通过结合图片内容和标签文本、描述文本对图片进行自动分类，可以提升图片分类的速度和准确度；通过基于空间注意力机制学习图像中重要的区域，对输入的图片数据进行增强处理，然后用于图片分类，进一步提升了图片分类准确率。

技术领域

本申请实施例涉及计算机技术领域，特别是涉及一种图片分类方法、装置、电子设备及计算机可读存储介质。

背景技术

在评论或在线购物等应用中，商家和用户上传了海量的图片用于展示商品或评论。而这些评论中蕴含着大量对于商家或用户有用的信息。例如，平台通过分析用户上传的关于某一商家的图片，搜集商家的用户满意度、商家环境、地理位置等信息，从而获得更多的商家数据。再例如，平台通过分析用户上传的关于某一商家的图片，向搜索端展示更加丰富的商家信息。但是，现有技术中，网络平台中用户或商家上述图片没有被分类存储、管理或运用，导致例如图片被杂乱地进行展示，对用户的参考价值不大。平台也无法从上述图片中获取到准确的商家信息。

可见，现有技术中需要一种对上传至网络平台的图片的分类方法。

发明内容

本申请实施例提供一种图片分类方法，能够实现图片准确分类，从而提升图片的信息利用率。

为了解决上述问题，第一方面，本申请实施例提供了一种图片分类方法，包括：

获取目标图片，以及所述目标图片的标签文本和描述文本；

基于空间注意力机制确定所述目标图片的图片内容的向量表示，以及，分别确定所述标签文本和所述描述文本的向量表示；

基于所述图片内容的向量表示、所述标签文本和所述描述文本的所述向量表示，对所述目标图片进行分类处理。

第二方面，本申请实施例提供了一种图片分类装置，包括：

图片信息获取模块，用于获取目标图片，以及所述目标图片的标签文本和描述文本；

特征编码模块，用于基于空间注意力机制确定所述目标图片的图片内容的向量表示，以及，分别确定所述标签文本和所述描述文本的向量表示；

分类处理模块，用于基于所述图片内容的向量表示、所述标签文本和所述描述文本的所述向量表示，对所述目标图片进行分类处理。

第三方面，本申请实施例还公开了一种电子设备，包括存储器、处理器及存储在所述存储器上并可在处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现本申请实施例所述的图片分类方法。

第四方面，本申请实施例提供了一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时本申请实施例公开的图片分类方法的步骤。

本申请实施例公开的图片分类方法，通过获取目标图片，以及所述目标图片的标签文本和描述文本；基于空间注意力机制确定所述目标图片的图片内容的向量表示，以及，分别确定所述标签文本和所述描述文本的向量表示；基于所述图片内容的向量表示、所述标签文本和所述描述文本的所述向量表示，对所述目标图片进行分类处理，有助于对上传至网络平台的图片进行准确分类。

上述说明仅是本申请技术方案的概述，为了能够更清楚了解本申请的技术手段，而可依照说明书的内容予以实施，并且为了让本申请的上述和其它目的、特征和优点能够更明显易懂，以下特举本申请的具体实施方式。

附图说明