摘要
本发明公开了基于自注意力网络的遥感图像城市景观分类方法及系统,涉及遥感图像识别分类技术领域,该方法包括:获取遥感图像和分类标记并进行数据增强;将数据集中的遥感图像分割,获得图像块序列;构建基于视觉Transformer模型的自注意力网络,包括嵌入层、编码器和任务头;输入图像块序列和分类标记至自注意力网络,在嵌入层中获得其嵌入的表示后与分类标记连接,获得带标记的结果嵌入图像块序列;将带标记的结果嵌入图像块序列输入编码器,生成特征表示和注意力分数;将特征表示输入任务头,并根据注意力分数生成分类标签,完成遥感图像城市景观分类。本发明泛化性能高,基于图像全局上下文依赖关系进行分类,提高了准确性。