【技术实现步骤摘要】
基于通道多分组融合的遥感场景图像分类方法
[0001]本专利技术涉及遥感场景图像分类方法。
技术介绍
[0002]遥感场景图像分类的目标是将输入的遥感图像进行正确的分类。由于遥感图像分类在自然灾害检测,土地覆盖分析,城市规划和国防安全[1,2,3,4]等方面的广泛应用,引起了大家的关注。到目前为止,已经提出了许多用于遥感场景图像分类的方法。其中,卷积神经网络凭借自身强大的特征提取能力,成为公认的最成功深度学习方法,被广泛应用于图像分类[5],目标检测[6]等方向。很多优秀的神经网络被设计用来进行图像分类。例如,Li等人[7]提出了一种用于遥感场景分类的深度特征融合网络。赵等人[8]提出了一个结合局部光谱特征、全局纹理特征和局部结构特征的PTM框架对特征进行融合。王等人[9]利用注意机制自适应地选择每个图像的关键部分,然后进行特征融合生成强大的特征。
[0003]近年来,设计一个在分类精度和运行速度之间实现最优权衡的卷积神经网络成为了一个热门的研究方向。SqueezeNet[10]通过挤压和扩展模块来降低参数量实现轻量级网络。挤压部分是由一组连续的1
×
1卷积组成,扩展部分是由一组连续的1
×
1卷积和一组连续的3
×
3卷积进行通道连接组成。MobileNetV1[11]引入了深度可分离卷积来替代传统的卷积,深度可分离卷积分成两个独立的过程:用于空间滤波的轻量级深度卷积和用于生成特征的1
×
1卷积,将空间滤波从特征产生机制中分离出来,有效的分 ...
【技术保护点】
【技术特征摘要】
1.基于通道多分组融合的遥感场景图像分类方法,其特征在于:所述方法具体过程为:步骤一、获取高光谱图像;步骤二、建立基于通道多分组融合的轻量级卷积神经网络模型;步骤三、将高光谱图像输入建立的基于通道多分组融合的轻量级卷积神经网络模型进行训练,得到训练好的基于通道多分组融合的轻量级卷积神经网络模型;所述基于通道多分组融合的轻量级卷积神经网络模型包括输入层、第一组、第二组、第三组、第四组、第五组、第六组、第七组、第八组和输出层;第一组和第二组分别包括分支1、分支2和分支3;分支1依次包括第一步长为2卷积核大小是3
×
3的卷积层、批标准化、Rule激活函数、第二步长为1卷积核大小是3
×
3的卷积层、批标准化、Rule激活函数;分支2依次包括步长为2池化核大小为2的最大池化层、第三步长为1卷积核大小是3
×
3的卷积层;分支3包括第四步长为1卷积核大小是1
×
1的卷积层;将分支1和分支2得到的特征进行融合;将融合之后的特征和分支3得到的特征进行特征融合,得到最终的输出特征;第三组依次包括第五步长为1卷积核大小是1
×
1的卷积层、第六步长为1卷积核大小是3
×
3的卷积层和第一步长为2卷积核大小是3
×
3的深度可分离卷积层;第四组至第七组分别包括第七卷积单元、第八卷积单元、第九卷积单元、第十卷积单元、第二深度可分离卷积单元、第三深度可分离卷积单元、第四深度可分离卷积单元、第五深度可分离卷积单元、第六深度可分离卷积单元、第七深度可分离卷积单元、第八深度可分离卷积单元、第九深度可分离卷积单元、第十深度可分离卷积单元;所述第七卷积单元依次包括第七卷积层、批标准化、Rule激活函数;所述第八卷积单元依次包括第八卷积层、批标准化、Rule激活函数;所述第九卷积单元依次包括第九卷积层、批标准化、Rule激活函数;所述第十卷积单元依次包括第十卷积层、批标准化、Rule激活函数;所述第二深度可分离卷积单元依次包括第二深度可分离卷积层、批标准化、Rule激活函数;所述第三深度可分离卷积单元依次包括第三深度可分离卷积层、批标准化、Rule激活函数;所述第四深度可分离卷积单元依次包括第四深度可分离卷积层、批标准化、Rule激活函数;所述第五深度可分离卷积单元依次包括第五深度可分离卷积层、批标准化、Rule激活函数;所述第六深度可分离卷积单元依次包括第六深度可分离卷积层、批标准化、Rule激活函数;所述第七深度可分离卷积单元依次包括第七深度可分离卷积层、批标准化、Rule激活函数;所述第八深度可分离卷积单元依次包括第八深度可分离卷积层、批标准化、Rule激活函数;
所述第九深度可分离卷积单元依次包括第九深度可分离卷积层、批标准化、Rule激活函数;所述第十深度可分离卷积单元依次包括第十深度可分离卷积层、批标准化、Rule激活函数;将通道数是C的输入特征划分成两个部分,一部分由4个通道数为的特征组成,另一部分由2个通道数为的特征组成;将通道数是的特征分别输入第七卷积单元、第八卷积单元、第九卷积单元、第十卷积单元进行卷积操作;将第七卷积单元卷积操作结果和第八卷积单元卷积操作结果进行通道融合,融合后特征通道数是将特征通道数是的特征输入第二深度可分离卷积单元进行卷积操作;将第八卷积单元卷积操作结果和第九卷积单元卷积操作结果进行通道融合,融合后特征通道数是将特征通道数是的特征输入第三深度可分离卷积单元进行卷积操作;将第九卷积单元卷积操作结果和第十卷积单元卷积操作结果进行通道融合,融合后特征通道数是将特征通道数是的特征输入第四深度可分离卷积单元进行卷积操作;将通道数是的特征分别输入第五深度可分离卷积单元和第六深度可分离卷积单元进行卷积操作;将第二深度可分离卷积单元的卷积操作结果和第三深度可分离卷积单元的卷积操作结果进行通道融合,融合后每个特征的通道数是C,将通道数为C的特征输入第七深度可分离卷积单元进行卷积操作;将第三深度可分离卷积单元的卷积操作结果和第四深度可分离卷积单元的卷积操作结果进行通道融合,融合后每个特征的通道数是C,将通道数为C的特征输入第八深度可分离卷积单元进行卷积操作;将第四深度可分离卷积单元的卷积操作结果和第五深度可分离卷积单元的卷积操作结果进行通道融合,融合后每个特征的通道数是C,将通道数为C的特征输入第九深度可分离卷积单元进行卷积操作;将第五深度可分离卷积单元的卷积操作结果和第六深度可分离卷积单元的卷积操作结果进行通道融合,融合后每个特征的通道数是C,将通道数为C的特征输入第十深度可分离卷积单元进行卷积操作;将第七深度可分离卷积单元输出结果、第八深度可分离卷积单元输出结果、第九深度可分离卷积单元输出结果、第十深度可分离卷积单元输出结果进行特征融合,特征融合结果和输入特征进行短连接得到输出特征;第八组依次包括全局平均池化层和SoftMax分类器;输入层连接第一组,第一组输出连接第二组,第二组输出连接第三组,第三组输出连接第四组,第四组输出连接第五组,第五组输出连接第六组,第六组输出连接第七组,第七组
输出连接第八组,第八组连接输出层;步骤四、将待测高光谱图像输入练好的基于通道多分组融合的轻量级卷积神经网络模型得到分类结果。2.根据权利要求1所述基于通道多分组融合的遥感场景图像分类方法,其特征在于:所述第一组和第二组分别包括分支1、分支2和分支3;分支1依次包括第一步长为2卷积核大小是3
×
3的卷积层、批标准化、Rule激活函数、第二步长为1卷积核大小是3
×
3的卷积层、批标准化、Rule激活函数;分支2依次包括步长为2池化核大小为2的最大池化层、第三步长为1卷积核大小是3
×<...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。