一种高光谱图像分类方法

文档序号:37282807发布日期:2024-03-12 21:23阅读:来源:国知局

技术特征:

1.一种高光谱图像分类方法,所述高光谱图像由高光谱相机拍摄获得,其特征在于,所述高光谱图像分类方法基于卷积神经网络和混合组合transformer注意力网络,具体包括以下步骤:

2.根据权利要求1所述的高光谱图像分类方法,其特征在于,所述步骤s1包括:

3.根据权利要求2所述的高光谱图像分类方法,其特征在于,步骤s2包括:

4.根据权利要求3所述的高光谱图像分类方法,其特征在于,所述步骤s3包括:

5.根据权利要求2所述的高光谱图像分类方法,其特征在于,所述步骤s4包括:

6.根据权利要求5所述的高光谱图像分类方法,其特征在于,所述步骤s5包括:

7.根据权利要求6所述的高光谱图像分类方法,其特征在于,所述步骤s6包括:

8.根据权利要求1所述的高光谱图像分类方法,其特征在于,所述步骤s7包括:

9.根据权利要求1-8中任一项所述的高光谱图像分类方法,其特征在于,所述高光谱图像分类方法使用一种高光谱图像分类系统完成,所述高光谱图像分类系统包括高光谱图像模块(1)、3d密集卷积神经网络模块(2)、基于光谱特征的混合组合transformer注意力网络模块(3)、2d密集卷积神经网络模块(4)、基于空间特征的混合组合transformer注意力网络模块(5)、光谱-空间特征融合网络模块(6)和分类图像模块(7),所述步骤s1在高光谱图像模块(1)中完成,所述步骤s2在3d密集卷积神经网络模块(2)中完成,所述步骤s3在基于光谱特征的混合组合transformer注意力网络模块(3)中完成,所述步骤s4、和所述步骤s5在2d密集卷积神经网络模块(4)中完成,所述步骤s6在基于空间特征的混合组合transformer注意力网络模块(5)中完成,所述步骤s7和所述步骤s8在光谱-空间特征融合网络模块(6)中完成,而步骤s9在分类图像模块(7)中完成。


技术总结
本发明公开了一种高光谱图像分类方法,其结合了卷积神经网络和混合组合变换器注意力网络的优点。在光谱特征提取方面,利用3D密集卷积神经网络获取高光谱图像的局部光谱特征信息,然后利用混合组合变换器注意力网络来同时捕捉各种组大小的token以及组之间的关联;在空间特征提取方面,利用2D密集卷积神经网络获取高光谱图像的局部空间特征信息,然后利用混合组合变换器注意力网络来增强高光谱图像的全局空间信息的利用。最后,将空间和光谱特征信息经特征融合网络实现空间‑光谱特征信息的融合,再经过Softmax层输出分类结果。本发明实现了对高光谱图像的准确分类,采用了轻量级Sophia优化器,可极大加快模型的运行速度。

技术研发人员:张绍泉,唐孟雄,梁联晖,李政奎,李璠,赖鹏飞,郑嘉焌,邓承志,汪胜前
受保护的技术使用者:南昌工程学院
技术研发日:
技术公布日:2024/3/11
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1