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