[发明专利]一种逐像素分类方法、存储介质及分类设备在审
申请号: | 202010819496.0 | 申请日: | 2020-08-14 |
公开(公告)号: | CN111967516A | 公开(公告)日: | 2020-11-20 |
发明(设计)人: | 马文萍;马梦茹;朱浩;武越;焦李成 | 申请(专利权)人: | 西安电子科技大学 |
主分类号: | G06K9/62 | 分类号: | G06K9/62;G06N3/04;G06N3/08 |
代理公司: | 西安通大专利代理有限责任公司 61200 | 代理人: | 高博 |
地址: | 710071 陕*** | 国省代码: | 陕西;61 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 一种 像素 分类 方法 存储 介质 设备 | ||
1.一种逐像素分类方法,其特征在于,包括以下步骤:
S1、从数据集中读入多光谱影像,包括已配准后的PAN图像数据和MS图像数据以及对应的类标ground truth图;
S2、将步骤S1得到的PAN图像数据和MS图像数据的共有特征进行融合得到MSHPAN图像数据;
S3、对步骤S1的PAN图像数据、MS图像数据和步骤S2的MSHPAN图像数据进行归一化处理,确定训练集和测试集;
S4、分别对步骤S1的PAN图像数据、MS图像数据和步骤S2的MSHPAN图像数据进行预处理,使PAN图像数据、MSHPAN图像数据和MS图像数据的特征块尺寸与通道数相同;
S5、设计自适应感受野网络模块A;
S6、设计自适应通道的谱段信息提取的B模块;
S7、利用步骤S4预处理后的PAN图像数据、MSHPAN图像数据和MS图像数据,设计基于步骤S5自适应感受野网络模块A和步骤S6自适应通道的谱段信息提取的B模块的渐进融合网络;
S8、构造步骤S7渐进融合网络的损失函数;
S9、对步骤S7渐进融合网络进行超参数调整;
S10、用步骤S3训练集的样本块对作为分类器的输入,MS图像数据、MSHPAN图像数据、PAN图像数据相对应的一对同时输入进融合网络的三个支路,将训练集中每个像素点的类别作为分类模型的输出,优化分类模块的网络参数,得到训练好的分类模型;
S11、利用步骤S10训练好的分类模型对步骤S3的测试集进行分类,得到测试数据集中每个像素点的类别。
2.根据权利要求1所述的方法,其特征在于,步骤S3具体为:
S301、数据归一化,分别对PAN、MSHPAN和MS图像矩阵中所有像素点的值归一化到[0,1]之间;
S302、选取训练集和测试集,在归一化后的图像中找到有对应类标的像素点,以每个像素为中心取PAN图和MSHPAN图邻域64*64的区域,MS图邻域16*16作为样本块对,每个块的邻域为中心点的分类所服务;然后取每类的百分之十样本作为训练集,其余作为测试集。
3.根据权利要求1所述的方法,其特征在于,步骤S5具体为:
S501、通过不同的卷积核输入进行卷积操作,分别得到F1、F2、F3,保留输入F0,将输入全连接的神经网络,最后通过sigmoid函数,将其约束在[0,1]之间,得到其掩膜α0、α1、α2、α3;
其中,c表示通道数;
S502、最后将掩膜与其对应的输入相乘得到
S503、通过其每个掩膜的均值选出两个影响最大的感受野获取到的特征(S1、S2)进行拼接得到S,这样可以自适应多视点的提取特征,提高空间信息的提取效率,进而提高分类的精度;S=S1+S2
S504、然后通过将S输入到sigmoid激活函数中得到自适应感受野的输出SF;
SF=σ(S)
其中,σ为sigmoid激活函数。
4.根据权利要求1所述的方法,其特征在于,步骤S6具体为:
S601、将于自适应感受野网络模块A的输出SF作为自适应通道的谱段信息提取的B模块的输入,将SF的维度均分成两份,然后通过全局平均池化压缩SF的空间信息,保留通道谱段信息得到VF;
其中,H和W为高和宽,Fgp为全局平均池化,(i,j)为特征SF的逐像素位置;
S602、通过softmax激活函数得到掩膜α和β,将得到的掩膜与其输入分别相乘得到和将其拼接作为自适应通道的谱段信息提取的B模块的输出O;
α+β=1
其中,为乘上权重α的特征,为乘上权重β的特征,α为S1的掩膜,β为为S2的掩膜。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于西安电子科技大学,未经西安电子科技大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202010819496.0/1.html,转载请声明来源钻瓜专利网。