一种基于双重对比提示学习的零样本图像分类方法

本发明提供了一种基于双重对比提示学习的零样本图像分类方法，包括：步骤1，拼接可学习向量与类别名构建可学习的prompt，并将其送入到CLIP的文本编码器中得到文本特征；步骤2，对图像数据进行数据增强，并将其送入到CLIP的图像编码器中得到图像特征；步骤3，将得到的文本特征和图像特征在文本空间和视觉空间上分别进行对比学习，训练得到prompt；步骤4，利用训练完成的prompt进行零样本图像分类，提升零样本图像分类性能。