基于SRCM和卷积神经网络的高光谱图像分类方法技术

技术编号:20820827 阅读:42 留言:0更新日期:2019-04-10 06:11
一种基于光谱反射率曲线矩阵SRCM和卷积神经网络的高光谱图像分类方法,主要解决现有技术中高光谱图像分类精度低的问题。本发明专利技术具体步骤如下:(1)构建卷积神经网络;(2)对待分类的高光谱图像加噪;(3)对图像数据逐波段归一化;(4)生成空间光谱特征矩阵集合;(5)生成堆叠空间光谱反射率曲线矩阵集合;(6)生成训练数据集和测试数据集;(7)训练卷积神经网络;(8)对测试数据集进行分类。本发明专利技术利用融合不同层特征的卷积神经网络对高光谱图像的堆叠空间光谱反射率曲线矩阵进行特征学习,从而进行分类,具有针对高光谱图像分类问题精度高的优点。

【技术实现步骤摘要】
基于SRCM和卷积神经网络的高光谱图像分类方法
本专利技术属于图像处理
,更进一步涉及目标识别
中一种基于光谱反射率曲线矩阵SRCM(SpectralReflectanceCurveMatrix)和卷积神经网络的高光谱图像分类方法。本专利技术可用于农业遥感、地图制图、环境监测、癌症检测和植被调查等领域进行地物目标识别。
技术介绍
高光谱图像能够同时描述地物空间分布的二维辐射信息和光谱信息,构成独特的空间、辐射和光谱三重信息合一的谱图,在遥感应用领域中得到广泛应用。分类是高光谱图像处理技术中的一项重要内容,其最终目标是给图像中的每个像元赋以唯一的类别标识,在军事和民用方面都有着重要的应用。近几年,深度学习在模式识别领域已经发挥出了极大的作用,为提高高光谱图像分类精度,深度学习中的堆栈自编码器、卷积神经网络、深度置信网络、生成对抗网络等深度模型被引入到高光谱图像分类中。YushiChen等人在发表的论文“DeepLearning-BasedClassificationofHyperspectralData”(IEEEJournalofSelectedTopicsinAppli本文档来自技高网...

【技术保护点】
1.一种基于光谱反射率曲线矩阵SRCM和卷积神经网络的高光谱图像分类方法,其特征在于,构建卷积神经网络,利用生成的堆叠空间光谱反射率曲线矩阵集合训练卷积神经网络,利用训练好的卷积神经网络进行分类;该方法的具体步骤如下:(1)构建卷积神经网络:(1a)构建一个20层的卷积神经网络,其结构依次为:输入层→第1个卷积层→第1个池化层→第2个卷积层→第2个池化层→第3个卷积层→第3个池化层→第4个卷积层→第4个池化层,第1个池化层→第5个卷积层→第1个全连接层,第2个池化层→第6个卷积层→第2个全连接层,第3个池化层→第7个卷积层→第3个全连接层,第4个池化层→第8个卷积层→第4个全连接层,第1个全连...

【技术特征摘要】
1.一种基于光谱反射率曲线矩阵SRCM和卷积神经网络的高光谱图像分类方法,其特征在于,构建卷积神经网络,利用生成的堆叠空间光谱反射率曲线矩阵集合训练卷积神经网络,利用训练好的卷积神经网络进行分类;该方法的具体步骤如下:(1)构建卷积神经网络:(1a)构建一个20层的卷积神经网络,其结构依次为:输入层→第1个卷积层→第1个池化层→第2个卷积层→第2个池化层→第3个卷积层→第3个池化层→第4个卷积层→第4个池化层,第1个池化层→第5个卷积层→第1个全连接层,第2个池化层→第6个卷积层→第2个全连接层,第3个池化层→第7个卷积层→第3个全连接层,第4个池化层→第8个卷积层→第4个全连接层,第1个全连接层→特征级联层,第2个全连接层→特征级联层,第3个全连接层→特征级联层,第4个全连接层→特征级联层,特征级联层→第1个Dropout层→第1个Softmax层;(1b)设置卷积神经网络各层参数如下:将输入层特征映射图总数设置为1;将第1个卷积层特征映射图总数设置为6、卷积核大小设置为5×5;将第1、2、3、4个池化层池化窗口都设置为2×2;将第2个卷积层特征映射图数目设置为12,卷积核大小设置为5×5;将第3个卷积层特征映射图数目设置为32,卷积核大小设置为4×4;将第4个卷积层特征映射图数目设置为64,卷积核大小设置为3×3;将第5、6、7、8个卷积层特征映射图数目都设置为1,卷积核大小都设置为1×1;将第1、2、3、4个全连接层节点总数都设置为64;将第1个Dropout层节点总数设置为256,置零概率为0.5;将第1个Softmax层节点总数设置为待分类的高光谱图像的地物类别总个数;(1c)特征级联层将卷积神经网络的第1、2、3、4个全连结层的输出按行排序组成融合特征矩阵;(2)对待分类的高光谱图像加噪:(2a)输入一幅待分类的高光谱图像;(2b)对待分类的高光谱图像中的每个像素点的每个波段的像素值,分别加入服从[-50,50]上的均匀分布的随机噪声,得到加噪后的高光谱图像;(3)对图像数据逐波段归一化:(3a)分别提取待分类的高光谱图像和加噪后的高光谱图像中每个波段的像素值的最大值和最小值;(3b)利用归一化公式,利用每个波段的像素值的最大值和最小值,分别对待分类的高光谱图像和加噪后的高光谱图像的每个波段中每个像素点进行归一化处理;(4)生成空间光谱特征矩阵集合:(4a)利用空间光谱特征矩阵提取方法,分别提取归一化后的待分类的高光谱图像和加噪后的高光谱图像的每个像素点的空间光谱特征矩阵;(4b)将归一化后的待分类的高光谱图像中所有像素点的空间光谱特征矩阵组成待分类的高光谱图像的空间光谱特征矩阵集合,将归一化后的加噪后的高光谱图像中所有像素点的空间光谱特征矩阵组成加噪后的高光谱图像的空间光谱特征矩阵集合;(5)生成堆叠空间光谱反射率曲线矩阵集合:(5a)将零矩阵的行数和列数分别初始化为待分类的高光谱图像的波段总数;(5b)利用矩阵更新方法,更新零矩阵中的元素值,将更新后的零矩阵分别作为待分类的高光谱图像和加噪后的高光谱图像的每个像素点的空间光谱特征矩阵中每列向量的光谱反射率曲线矩阵;(5c)利用堆叠矩阵计算公式,分别计算待分类的高光谱图像和加噪后的高光谱图像的空间光谱特征矩阵集合中每个像素点的堆叠空间光谱反射率曲线矩阵;(5d)将待分类的高光谱图像中所有像素点的堆叠空间光谱反射率曲线矩阵,组成待分类的高光谱图像的堆叠空间光谱反射率曲线矩阵集合,将加噪后的高光谱图像中所有像素点的堆叠空间光谱反射率曲线矩阵,组成加噪后的高光谱图像的堆叠空间光谱反射率曲线矩阵集合;(6)生成训练数据集和测试数据集:分别从待分类的高光谱图像和加噪后的高光谱图像的堆叠空间光谱反射率曲线矩阵集合中,随机选取80%的有标记的像素点的堆叠空间光谱反射率曲线矩阵组成训练数据集,将其余待分类的高光谱图像的堆叠空间光谱反射率曲线矩阵集合中的有标记的像素点的堆叠空间光谱反射率曲线矩阵作为测试数据集;(7)训练卷积神经网络:利用随机梯度下降方法,对卷积神经网络进行2000次训练,得到训练好的卷积神经网络;(8)对测试数据集进行分类:分别将测试数据集中的每个测试样本输入到卷积神经网络中,通过第1个Softmax层上的Softmax分类器,得到每个测试样本的分类标签,作为对应像素点的分类结果。2.根据权利要求1中所述的基于光谱反射率曲线矩阵SRCM和卷积神经网络的高光谱图像分类方法,其特...

【专利技术属性】
技术研发人员:王桂婷李诗卉公茂果钟桦吴飞杨晓婕陈贝贝马锐解玮
申请(专利权)人:西安电子科技大学
类型:发明
国别省市:陕西,61

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1