[发明专利]一种图像分类方法、装置、终端及存储介质

摘要

本发明公开了一种图像分类方法、装置、终端及存储介质，方法通过获取待分类图像，将待分类图像输入目标分类模型，其中，目标分类模型包括卷积层、注意力层以及分类层；通过卷积层获取待分类图像的局部特征信息，得到第一特征图；通过注意力层对第一特征图进行全局建模，得到第二特征图；通过分类层对第二特征图进行图像分类，得到待分类图像对应的图像类别。本发明中的目标分类模型可以捕捉待分类图像的局部特征信息也可以进行全局建模，因此可以精确地预测出待分类图像对应的图像类别。解决了现有技术中深度卷积神经网络仅具备有捕捉局部上下文信息的能力，而不具备全局建模的能力，导致模型的分类性能不佳的问题。