一种基于重叠池化的高光谱遥感影像分类方法技术

本发明专利技术公开了一种基于重叠池化的高光谱遥感影像分类方法，包括以下步骤：S1：选取高光谱遥感图像数据集；S2：对步骤S1得到的高光谱遥感图像数据集进行预处理；S3：确定感兴趣的地物类别数，构建训练样本集和测试集；S4：构建基于重叠池化的卷积神经网络模型；S5：用步骤S4所构建的模型完成高光谱数据分类，并对分类结果进行分析评价。本发明专利技术提高了采样的准确性和网络分类精度，并在一定程度上减少了过拟合问题。

【技术实现步骤摘要】
一种基于重叠池化的高光谱遥感影像分类方法
本专利技术涉及高光谱影像分类技术，特别是涉及一种基于重叠池化的高光谱遥感影像分类方法。
技术介绍
高光谱影像(HSI)数据是同时包括空间和数百个光谱带信息在内的持续数据。利用丰富的光谱信息，高光谱数据分类已经应用在诸如土地利用，矿物勘探，水污染检测等诸多方面。其在相同的空间区域内包含数百个光谱带的特点，可以获得连续的光谱值并分辨许多不同的地物，为识别各种材料提供了有价值的信息。然而，由于在某些波段频谱范围内不同地物之间光谱值差别很小，并且传统方式往往省略了空间特征，这就增加高光谱分类难度。近来，由于深度学习模型可以学习越来越复杂的深层次特征结构，基于深度学习的遥感影像分析方法越来越受到重视，并被广泛应用于HSI应用领域。卷积神经网络(CNN)由交替堆叠的卷积层和池化层组成。卷积层是通过线性卷积滤波器提取特征图。池化层是将相邻像素的空间特征组合在一起，用于实现图像变换的不变性，提高物体轻微变形的鲁棒性。CNN具有本地连接和共享权重的优点，使其有更少的链接、参数，且更容易训练。另外，其学习能力可以通过控制网络的深度和宽度来调整。因此，CNN凭借其对HSI的光谱及空间特征的强大提取能力，在高光谱影像上的分类应用越来越多，且取得不错的成绩。然而，CNN模型的池化方式多采用非重叠池化，忽略HSI中邻近像素对特征提取的影响，造成模型分类精度受到限制。
技术实现思路
专利技术目的：本专利技术的目的是针对现有基于非重叠池化的CNN模型在HIS分类应用上的不足，提出一种基于重叠池化的高光谱遥感影像分类方法。技术方案：本专利技术所述的基于重叠池化的高光谱遥感影像分类方法，包括以下步骤：S1：选取高光谱遥感图像数据集；S2：对步骤S1得到的高光谱遥感图像数据集进行预处理；S3：确定感兴趣的地物类别数，构建训练样本集和测试集；S4：构建基于重叠池化的卷积神经网络模型；S5：用步骤S4所构建的模型对步骤S2预处理得到的数据集进行分类，并对分类结果进行分析评价。进一步，所述步骤S2中，预处理为：将高光谱遥感图像数据集中每一个一维形式的像素点光谱矢量转化为二维矩阵的形式。使得步骤S4中的卷积网络不仅可以学习到数据集的光谱特征，还能学习到其空间特征，从而提高网络整体性能，有利于分类精度的提高。进一步，所述步骤S4中，卷积神经网络模型包括Block1、Block2、Block3和Block4这4个块，以及1个全连接层。进一步，其中，Block1包括卷积层、最大池化层和修正线性激活层，Block2包括卷积层、ReLU层和平均池化层，Block3包括卷积层、ReLU层和平均池化层，Block4包括卷积层和ReLU层。据相关理论，特征提取的误差主要来自两个方面：邻域大小受限造成的估计值方差增大；卷积层参数误差造成估计均值的偏移。而最大池化可以减小第二种误差，平均池化能减小第一种误差，将两种池化方式组合即重叠池化，能够提高分类精度和加速特征的学习，并在一定程度上减轻过拟合问题。进一步，所述步骤S5中，采用Softmax-Loss分类模型完成数据集的分类，采用总体分类精度和卡帕系数来评估模型的分类性能。该模型具有计算步骤少、计算量小的特点，使得输出数值更为稳定，有利于分类精度的提高。有益效果：本专利技术公开了一种基于重叠池化的高光谱遥感影像分类方法，与现有技术相比，本专利技术具有如下有益效果：1)在CNN模型中，通过引入重叠池化操作，在池化输出特征中加入了邻近像素的空间维度信息，提高采样的准确性和网络分类精度，并在一定程度上减少了过拟合问题；2)设计出深度高达13层的深度卷积神经网络结构，加入ReLU层和Softmax-Loss分类模型，以提高网络模型性能，对于多种高光谱数据集都有高性能表现；3)将一维的像素点光谱矢量转换为二维形式输入CNN模型，使得不同标签之间的差异更为明显，对高光谱数据的空间特征的提取更为准确，从而大幅提高模型即本专利技术的分类精度。附图说明图1为本专利技术具体实施方式的神经网络结构的示意图；图2为本专利技术具体实施方式中步骤S2的数据预处理的示意图。具体实施方式下面结合具体实施方式和附图，对本专利技术的技术方案做进一步的介绍。本具体实施方式公开了一种基于重叠池化的高光谱遥感影像分类方法，其神经网络结构如图1所示，包括以下步骤：S1：选取高光谱遥感图像数据集。本实施例选取了两个数据集，其一为AVIRIS(AirborneVisible/InfraredImagingSpectrometer)于1992年6月在印第安纳州西北部拍摄的一幅高光谱遥感影像，影像是一块有农田与树林混合的遥感实验区，即IndianPines测试区。影像尺寸为145×145，空间分辨率为每像素20米，在丢弃受大气吸收影响的四个零频带和20个较低的信噪比(SNR)频带之后，保留了200个光谱带，数据集包含10249个标记的像素和16个真实地物类别，其中大部分是不同类型的农作物。其二为Salinas数据集，它是由AVIRIS传感器在加利福尼亚谷的Salinas收集的，空间分辨率为每像素3.7米，该区域包括512×217像素，除去20个吸水带后，保留224个带中的204个光谱带，地面实况由54129个像素和16个土地覆盖类别组成，包括蔬菜，裸露土壤和葡萄园。S2：对步骤S1得到的高光谱遥感图像数据集进行预处理。如图2所示，把一维光谱矢量数据转换为二维矩阵形式的数据，IndiaPines数据与Salinas数据的输入尺寸均为14*14S3：确定感兴趣的地物类别数，构建训练样本集和测试集。分别取Indianpines数据集和Salinas数据集中10％的标记样本作为训练样本集，其余标记样本为测试集，如表1-2所示。表1IndianPines数据集的土地覆盖类型、训练集、测试集表2Salinas数据集的土地覆盖类型、训练集、测试集S4：构建基于重叠池化的卷积神经网络模型。如图1为模型的网络结构，卷积神经网络模型包括Block1、Block2、Block3和Block4这4个块，以及1个全连接层。其中，Block1包括卷积层、最大池化层和修正线性激活层，Block2包括卷积层、ReLU层和平均池化层，Block3包括卷积层、ReLU层和平均池化层，Block4包括卷积层和ReLU层。在模型中使用随机梯度下降法来训练网络，输入批次大小为100，迭代次数为100次，权值衰减为0.0001。权值衰减可以有效的抑制网络的过拟合问题，权值衰减惩罚项使得权值收敛到较小的绝对值，而惩罚大的权值，因为大的权值会使得系统出现过拟合，降低其泛化性能。此外，对学习率采用梯度下降的方式进行设置：[0.05*500.005*300.0005*20]，随着训练批次增加，学习率也随之下降一个数量级。S5：用步骤S4所构建的模型对步骤S2预处理得到的数据集进行分类，并对分类结果进行分析评价。采用Softmax-Loss分类模型完成数据集的分类，并用总体精度(OA)和卡帕系数(Kappa)来评估模型分类性能。实验结果如表3所示。表3重叠池化与非重叠池化分类性能由上表可知，重叠池化结构相比较于传统的非重叠池化会带来更高的特征提取精度和更快的网络学习速度。IndianPines数据准确率提高本文档来自技高网...

【技术保护点】
一种基于重叠池化的高光谱遥感影像分类方法，其特征在于：包括以下步骤：S1：选取高光谱遥感图像数据集；S2：对步骤S1得到的高光谱遥感图像数据集进行预处理；S3：确定感兴趣的地物类别数，构建训练样本集和测试集；S4：构建基于重叠池化的卷积神经网络模型；S5：用步骤S4所构建的模型对步骤S2预处理得到的数据集进行分类，并对分类结果进行分析评价。

【技术特征摘要】
1.一种基于重叠池化的高光谱遥感影像分类方法，其特征在于：包括以下步骤：S1：选取高光谱遥感图像数据集；S2：对步骤S1得到的高光谱遥感图像数据集进行预处理；S3：确定感兴趣的地物类别数，构建训练样本集和测试集；S4：构建基于重叠池化的卷积神经网络模型；S5：用步骤S4所构建的模型对步骤S2预处理得到的数据集进行分类，并对分类结果进行分析评价。2.根据权利要求1所述的基于重叠池化的高光谱遥感影像分类方法，其特征在于：所述步骤S2中，预处理为：将高光谱遥感图像数据集中每一个一维形式的像素点光谱矢量转化为二维矩阵的形式。3.根据权利要求1所述的基于重叠池化的高光谱遥感影像分类方法，...

【专利技术属性】
技术研发人员：高红民，杨耀，李臣明，蔺硕，周惠，王建华，张振，樊悦，
申请(专利权)人：河海大学，
类型：发明
国别省市：江苏,32