基于视觉Transformer的遥感图像分类方法

    公开(公告)号:CN117746141A

    公开(公告)日:2024-03-22

    申请号:CN202311775891.3

    申请日:2023-12-21

    Abstract: 本发明属于遥感图像分类领域,涉及基于视觉Transformer的遥感图像分类方法,包括:获取遥感图像并进行预处理,将预处理后的结果输入分块嵌入模块,得到特征矩阵;将特征矩阵拉长得到初始序列,在初始序列中加入类别token向量,得到类别序列;将类别序列输入Transformer编码模块,得到编码序列;将特征矩阵输入空间结构计算模块,得到空间结构序列;将编码序列和空间结构序列对应相加,得到结合序列,将结合序列输入全局结构融合模块,得到融合序列;将融合序列的类别token向量输入分类头,得到分类结果;本发明采用多方向对称卷积与空间结构计算策略,从底层提高模型旋转鲁棒性,使其分类呈不同方向的遥感图像。

Patent Agency Ranking