有效

零样本图像分类方法、系统、电子设备及存储介质

李硕豪、王风雷、张军、练智超、雷军、李小飞、蒋林承、何华、李千目
中国人民解放军国防科技大学

摘要

本公开提供一种零样本图像分类方法、系统、电子设备及存储介质,该方法先提取输入图像的全局特征,然后基于注意力机制对所述全局特征进行学习,以得到多个特征掩码,基于所述多个最大掩码值中的最大值和预设的自适应因子,计算自适应阈值;基于所述自适应阈值与所述多个特征掩码得到所述输入图像的加权全局特征;计算所述加权全局特征与该未见类别的语义嵌入向量的相容性得分,确定所述多个相容性得分中的最大值作为最高相容性得分,并输出与所述最高相容性得分对应的所述未见类别作为对所述输入图像的类别预测结果,从而通过阈值自适应的注意力机制,抑制冗余特征的同时提高了特征的鲁棒性,进一步提高了分类的准确度。