一种图像分类方法、装置、计算机设备及存储介质

本发明涉及人工智能技术领域，公开了一种图像分类方法、装置、计算机设备及存储介质，包括：构建包含可变形注意力机制的Transformer模型；获取的公开数据集，对公开数据集进行预处理，得到目标数据集；基于目标数据集Transformer模型进行训练，得到图像分类模型；获取待分类的初始图像，对初始图像进行预处理，得到待分类的目标图像；将所待分类的目标图像输入图像分类模型，获取图像分类模型的输出结果，根据输出结果完成图像分类。本发明中的可变形注意力模块允许模型仅关注关键的局部区域，能够更有效地捕捉到图像中的重要局部特征，提高分类精度，而且在训练数据较少的情况下，通过关注重要特征来提升模型的泛化能力。