基于注意力机制和循环神经网络的高光谱图像分类方法

作者:冯婕; 赵宁; 焦李成; 张向荣; 尚荣华; 王蓉芳; 刘若辰
来源:2020-08-25, 中国, ZL202010860393.9.

摘要

本发明公开了一种基于注意力机制和循环神经网络的高光谱图像分类方法,主要解决现有技术中对高光谱图像特征提取不充分导致的分类精度偏低的问题。实现步骤为:(1)构造循环神经网络;(2)生成训练集;(3)训练循环神经网络;(4)对待分类的高光谱图像进行分类。本发明利用注意力提议模块和处理模块得到输入高光谱图像空间像素块需要重点关注的局部区域,利用循环神经网络提取待分类高光谱图像的空间-光谱联合特征,得到分类结果,具有高光谱图像分类精度高的优点,可用于资源勘探、森林覆盖和灾害监测等领域的地物目标识别。