基于双分支网络的光谱-空间注意力机制的高光谱图像分类方法

    公开(公告)号:CN117218429A

    公开(公告)日:2023-12-12

    申请号:CN202311178205.4

    申请日:2023-09-13

    申请人: 安徽大学

    摘要: 本发明涉及一种基于双分支网络的光谱‑空间注意力机制的高光谱图像分类方法,包括:输入高光谱图像数据立方体;使用主成分分析将高光谱图像的光谱维数降维,对于要分类的像素,将其封装成相邻区域块作为光谱子网络的输入;在每个特定像素周围创建一个相邻区域来收集空间信息,并作为空间子网络的输入;得到一维光谱特征;得到一维空间特征;通过融合层对一维光谱特征和一维空间特征进行融合和平衡,使用softmax回归层来预测每一类地物的概率分布。本发明通过对高光谱图像作为研究对象,在有限样本的前提下提高高光谱图像的有效特征表示,既保证了分类精度,又减少了网络模型的参数个数和计算量;充分利用了光谱信息和空间信息,以取得更好的分类结果。