一种基于深度学习的高光谱图像分类方法技术

本发明专利技术公开了一种基于深度学习的高光谱图像分类方法，包括预训练阶段和微调阶段，在对深度卷积网络进行改进的基础上对深层网络参数和浅层网络参数进行调整，能够在尽可能少的标签样本的情况下避免过拟合现象，提高了深层网络在小样本情况下的分类效果，在采用同等数据量进行训练网络时有一定的优势。

A Hyperspectral Image Classification Method Based on Deep Learning

【技术实现步骤摘要】
一种基于深度学习的高光谱图像分类方法
本专利技术涉及高光谱图像分类方法，特别是涉及一种基于深度学习的高光谱图像分类方法。
技术介绍
高光谱图像同时包含了空间信息和光谱信息，光谱信息的分辨率很高，一般都达到了纳米级别，空间分辨率相对较低，具体而言分类工作就是对图像中的每一个像素进行分类确认像素的类别。高光谱图像分类大体上可以分为无监督分类和有监督分类(包含半监督分类)两种。无监督分类指的是在没有预先标注数据标签的情况下对高光谱图像进行分类(聚类)，主要思想是依照能代表像素点的特征信息(空间信息、谱信息及特征等)将相似的像素归为一类。有监督的分类指的是在有预先标注数据作为监督信号的情况下对高光谱图像进行分类，主要思想是利用有标注的数据学习像素特征信息与像素类别之间的内在关系，然后利用这种关系对没有标注的数据进行分类，确定像素类别。目前基于深度学习的高光谱图像分类主要属于有监督分类，在高光谱领域应用的深度学习模型有卷积神经网络(CNN)、基于自编码网络的栈式自编码网络(SAE)、基于受限玻尔兹曼机的深度置信网络(DBN)等。深度学习的相关模型为高光谱图像分类中存在的问题提供了一些解决思路，本文档来自技高网...

【技术保护点】
1.一种基于深度学习的高光谱图像分类方法，其特征在于：包括以下步骤：S1：对一个已有的高光谱图像的数据集进行分割，得到源数据集、微调数据集和测试数据集；S2：对深度卷积网络的参数进行初始化；S3：利用源数据集训练深度卷积网络，将测试数据集输入到训练好的深度卷积网络中，得到测试数据集的正确率；S4：判断测试数据集的正确率是否大于第一阈值：如果是，则进行步骤S6；否则，则进行步骤S5；S5：调整深度卷积网络的参数，然后返回步骤S4；S6：将此时的浅层网络参数作为调整后的浅层网络参数；所述浅层是指深度卷积网络中用于提取所述已有的高光谱图像的数据集边缘和轮廓的卷积层；S7：对深层网络参数进行初始化；所...

【技术特征摘要】
1.一种基于深度学习的高光谱图像分类方法，其特征在于：包括以下步骤：S1：对一个已有的高光谱图像的数据集进行分割，得到源数据集、微调数据集和测试数据集；S2：对深度卷积网络的参数进行初始化；S3：利用源数据集训练深度卷积网络，将测试数据集输入到训练好的深度卷积网络中，得到测试数据集的正确率；S4：判断测试数据集的正确率是否大于第一阈值：如果是，则进行步骤S6；否则，则进行步骤S5；S5：调整深度卷积网络的参数，然后返回步骤S4；S6：将此时的浅层网络参数作为调整后的浅层网络参数；所述浅层是指深度卷积网络中用于提取所述已有的高光谱图像的数据集边缘和轮廓的卷积层；S7：对深层网络参数进行初始化；所述深层是指深度卷积网络中除浅层以外的其他卷积层；S8：利用微调数据集对步骤S6得到的调整后的浅层网络参数进行训练；S9：利用待分类的高光谱图像的数据集训练深度卷积网络；S10：输出结果；所述步骤S1-S5为预训练阶段，步骤S6-S10为微调阶段。2.根据权利要求1所述的基于深度学习的高光谱图像分类方法，其特征在于：所述步骤S4中的第一阈值为90％。3.根据权利要求1所述的基于深度学习的高光谱图像分类方法，其特征在于：所述步骤S1具体包括以下步骤：S11：选取IndianPines数据集作为已有的高光谱图像的数据集，数据集中数据的维数为145*145*220，数据集中共有16类作物；S12：去除数据集中20个水吸收严重的波段，将数据集中的数据进行维度转化；S13：随机选取已有的高光谱图像的数据集中25％的数据作为源数据集对深度卷积网络进行预训练，再从India...

【专利技术属性】
技术研发人员：高红民，姚丹，俞科栋，李臣明，杨耀，王明霞，缪雅文，
申请(专利权)人：河海大学，
类型：发明
国别省市：江苏,32