[发明专利]一种图像分类方法、装置、电子设备及存储介质

摘要

本公开关于一种图像分类方法、装置、电子设备及存储介质，包括：获取待分类图像；将所述待分类图像输入至预设多层神经网络中进行特征提取，得到每层神经网络的输出结果；将多个预设层神经网络的输出结果输入至多标签分类深度学习模型进行自注意力计算，得到所述待分类图像属于每个类别的概率。这样，结合待分类图像在每层神经网络的特征提取结果进行图像分类，由于底层神经网络的特征提取结果中往往包含更多的局部细节信息，因此，图像分类的结果也会参考到这些局部细节信息，减少局部细节信息经过多层卷积之后消失的情况，从而可以实现对待分类图像中的小目标的精准识别，提高多标签图像分类任务的精准度。