本发明涉及高光谱遥感图像分类方法领域,特别是一种结合快速区域生长超像素分割的高光谱遥感图像分类方法。
背景技术:
高光谱遥感图像分类作为地球遥感的一项关键技术,受到越来越多的人们关注,其应用于环境监测、城市测绘、精准农业等多个领域。近几十年来,利用高光谱图像的光谱信息,引入了大量的图像分类技术,典型的方法有多项逻辑回归,随机森林,稀疏表示和支持向量机等。随着高光谱遥感图像的空间分辨率得到了显著提高,研究人员尝试利用空间信息提高分类精度,其中超像素技术被广泛应用于高光谱图像的空间信息提取。超像素是由颜色和纹理特征相似的像素组成的区域,生成超像素的算法大致可以分为基于图和基于梯度上升的方法。超像素分割技术在高光谱图像分类方法中得到了广泛的应用,一定程度上提高了分类的结果。然而,目前基于超像素的高光谱图像分类仍然存在两个主要的挑战:(1)现有的超像素算法大多是针对二维的自然图像处理,对于三维的高光谱图像,超像素算法的性能可能会下降;(2)传统的基于超像素的高光谱分类方法一般将每个分割后的超像素作为一个单元进行图像分类,这可能会提高算法的效率,但通常会降低分类的精度。
技术实现要素:
本发明的目的在于提供一种结合快速区域生长超像素分割的高光谱遥感图像分类方法,它能有效的提高高光谱图像的分类精度。
实现本发明目的的技术解决方案为:一种结合快速区域生长超像素分割的高光谱遥感图像分类方法,该方法基于快速区域生长的超像素分割与用于高光谱图像分类的纹理自适应的融合策略,包括以下步骤:
步骤1、对高光谱图像进行特征提取,使用主成分分析的降维方法,计算各个波段对给定主成分的贡献率,得到前三个贡献率高的主成分图像,合成假彩色图像;
步骤2、在假彩色图像中初始化聚类中心,通过欧式距离计算聚类中心与未标签像素点的空间距离和光谱距离,并进行归一化,得到超像素与其邻域的未标签像素点相似性距离;
步骤3、对每个超像素进行区域生长,并构建距离映射图,对未标签像素点进行相似性距离与距离映射的比较,生成超像素;
步骤4、选取高光谱遥感图像,对每一类样本按照相同比例提取像素作为训练样本,进行基于像素的高光谱分类;
步骤5、对生成的每个超像素计算纹理自适应阈值;
步骤6、根据高光谱图像分类的纹理自适应融合策略确定每个像素的类别完成分类;其中所述融合策略是在基于像素的分类图中,当最大类别占比大于该超像素的自适应阈值时,将整个超像素内的类别均设为最大占比所对应的类别,否则不进行操作。
其中,在步骤2中,按照设定的超像素个数,在合成的假彩色图像内均匀的分配聚类中心点;假设图像总共有n个像素点,预分割为k个相同尺寸的超像素,那么每个超像素的大小为n/k,则相邻种子点的距离近似为s=sqrt(n/k),聚类中心为c[k]={xk,sk},由空间坐标xk和光谱值sk构成,并用新的元素表示ek={xk,sk,k,dk,k},其中k为超像素的序号,dk,k为第k的聚类中心与第k个超像素的相似性距离,使这些元素进入优先队列;在超像素的四邻域或八邻域通过欧式距离计算未标签像素与超像素的空间相似性和光谱相似性,并进行归一化;假设四邻域或八邻域其中一个像素表示为aj={xj,sj,k,dj,k},则与第k个超像素的相似性距离为:
其中空间相似性为:
光谱相似性为:
在步骤3中,为减少未标签像素的遍历次数,定义一个存取像素与超像素的最短距离矩阵d,在步骤2中计算的相似性距离dj,k后,若dj,k<d(xj),将更新d(xj)并且该元素进入优先队列;当队首元素出队时,该像素的分割标签确定,直至优先队列为空,即所有的元素的分割标签确定。
在步骤4中,对包含多类的高光谱图像,按照一定百分比提取每一类图像像素作为训练样本,采用基于像素的高光谱分类方法支持向量机进行分类,得到基于像素的高光谱分类图。
在步骤5中,对生成的超像素,使用对应假彩色图像中的光谱值方差vk来表示第k超像素的纹理复杂程度,第k超像素的纹理自适应阈值δk为vk与所有超像素方差的最大值max(v)的比值,表示如下:
与现有技术相比,本发明的显著优点在于:在对高光谱遥感图像通过区域生长生成超像素时,其中距离映射图的构造,在保证超像素分割的准确性之下,减少了超像素生成过程中像素遍历次数,加速了超像素的生成,并通过纹理自适应融合策略,提高了高光谱分类的精度,突破了传统的基于超像素的高光谱分类方法中,由于将超像素当作一个单元造成分类精度不高的问题。
附图说明
图1是本发明结合快速区域生长超像素分割的高光谱遥感图像分类方法的流程图。
图2是本发明结合快速区域生长超像素分割的高光谱遥感图像分类方法的结构框图。
图3是本发明在高光谱三维图像salinas-ascene与indian_pines上的基于区域生长分割后的超像素图。依此为高光谱三维图像的成像立方体,前三个主成分图像,生成的假彩色图像以及分割后的超像素图。
图4与图6分别是高光谱三维图像salinas-ascene与indian_pines的成像立方体,及标准类别分块图。左为三维成像立方体,右为标准类别分块图。
图5与图7分别是高光谱三维图像salinas-ascene与indian_pines的图像6类标签色块图例及按照10%比例提取的训练样本与测试样本数目示例图。
图8是本发明与其他方法分类结果的比较示例图。
具体实施方式
下面结合附图和实施例对本发明做进一步描述。
本发明以高光谱三维图像作为输入,本发明实施示例的流程如图1所示,结构框图如图2所示。
(1)在gic(grupodeinteligenciacomputacional)高光谱场景图像网站上下载得到的高光谱三维图像salinas-ascene与indian_pines大小分别为86×83×224与145×145×224,通过去除覆盖吸水区域的波段将波段数量减少到204个与220个,即校正后的可用图像大小分别为86×83×204与145×145×200;图4与图6给出了高光谱三维图像salinas-ascene与indian_pines的成像立方体及标准类别分块图。左为三维成像立方体,右为标准类别分块图,分别为6类和16类。
(2)分别对于校正后的高光谱三维数据salinas-ascene与indian_pines,通过特征提取,即主成分分析,得到高光谱图像的前三个主成分图像,并进行假彩色合成。
(3)分别设定超像素的个数为100与800,对两个假彩色图像进行初始化,确定聚类中心的空间坐标和光谱值,利用此信息和与超像素的距离表示为新的元素,进入优先队列。为了减少未标签像素的遍历次数,构造一个存取像素与超像素的最短距离的距离映射图,判断超像素邻域内未标签像素是否进入优先队列。当队内未标签像素出队时,该像素的分割标签得到确定。直至优先队列为空,即所有的元素的分割标签确定。图3给出了高光谱三维图像salinas-ascene与indian_pines基于区域生长之后的超像素分割图。依此为高光谱三维图像的成像立方体,前三个主成分图像,生成的假彩色图像以及分割后的超像素图。
(5)对于一个大小为86×83×204的高光谱三维数据salinas-ascene,将大小为86×83的二维投影图像上的每一个像素点视为一个样本,且每一个样本具有204维特征,结合自带类标签(groundtruth),从每一类中随机选取10%的样本构成训练集,剩余样本作为测试集。进行基于像素的高光谱分类,即支持向量机,得到基于像素的高光谱分类图。对于高光谱三维图像indian_pines进行相同的比例提取训练集。图5与图7分别给出了高光谱三维图像salinas-ascene与indian_pines6类和16类标签色块图例及按照10%的比例提取的训练样本与测试样本数目示例图。
(6)对于生成的每个超像素,为表示纹理复杂程度,对超像素求方差,并进行与最大的方差求比例的归一化操作,得到每个超像素的纹理自适应阈值。
(7)对于每个超像素对应的高光谱分类标签,计算最大类别所占比例,如果比例大于该超像素的自适应阈值,则将整个超像素中的所有像素标签设置为该类标签。否则超像素纹理较复杂,自适应阈值过大,无法改变超像素分类标签。完成所有测试样本的类别判定即获得整个高光谱三维图像salinas-ascene与indian_pines的分类结果。
(8)图8给出了本发明的分类结果与基于像素的分类方法的结果的比较示例图,其中第一列为标准分类结果,第二列为基于像素的高光谱分类结果,第三列为本发明分类结果。
本发明实现了高光谱图像分类精度的提高,在使用区域生长生成超像素时,通过构造距离映射图,减少了超像素生成过程中像素遍历次数,加速了超像素的生成,并通过纹理自适应融合策略,提高了高光谱分类的精度,突破了传统的基于超像素的高光谱分类方法中,将超像素当作一个单元所造成分类精度不高的问题。本发明获得的平均分类精度基本优于基于像素的分类方法,实现了高光谱图像分类的精度的提升,对于环境管理、农作物监测、矿物填图等方面具有重要的实际意义。