一种基于快速全卷积神经网络的SAR目标分类方法技术

技术编号:26033094 阅读:31 留言:0更新日期:2020-10-23 21:11
本发明专利技术提供一种基于快速全卷积神经网络的SAR目标分类方法,包括以下步骤:S1:对MASTAR数据集进行扩充S2:搭建步进≥2的特殊卷积层S3:搭建步进为2的全卷积神经网络S4:搭建步进为3的全卷积神经网络S5:搭建步进为4的全卷积神经网络S6:搭建快速全卷积神经网络根据步骤S3、步骤S4和步骤S5,采用3层卷积层进行特征提取、3层特殊卷积层进行下采样和2层全连接层分类的搭建。通过引入特殊卷积层进行下采样,可以大幅度提高卷积神经网络的训练效率,大大的改善现有SAR技术识别分类方面的困难,提高SAR图像的识别效率和准确度。

【技术实现步骤摘要】
一种基于快速全卷积神经网络的SAR目标分类方法
本专利技术属于电子信息
,具体地说,本专利技术涉及一种快速全卷积神经网络的SAR目标分类方法。
技术介绍
随着合成孔径雷达(SAR)技术的成熟,人们能够获取的SAR图像数据越来越多。然而,SAR目标的解译一直是一个世界性难题,至今没有得到很好的解决。如今,SAR解译已经形成了一套比较成熟的体系,但是,仍然存在以下几个问题:1、需要大量的先验知识;2、识别的准确度不稳定;3、当应用场景特别大时,计算量庞大。近年来,随着机器学习和人工智能的快速发展,很多学者开始考虑利用卷积神经网络(CNN)来对SAR目标进行分类识别。但是,现有的SAR图像数据远远不能满足CNN的训练需求,与此同时,训练CNN模型需要大量的时间也制约了该技术的发展。针对于传统CNN网络架构训练时间长的问题,快速全卷积神经网络(fasterACN)在保证识别准确度不明显下降的情况下能够减少大量的训练时间。通常,CNN模型一般由卷积层、池化层、全连接层等组成。卷积层负责特征提取;池化层负责下采样;全连接层负责最后的分类计算。这种固定的架构模式决定了CNN需要大量的训练数据是训练时间。鉴于此,本专利技术提出来一种快速全卷积神经网络(fasterACN)SAR目标分类方法。该方法的优势在于能够减少大量的模型训练时间,同时保持较高的识别准确度。
技术实现思路
针对现有技术存在的不足,本专利技术目的是提供一种基于快速全卷积神经网络(fasterACN)SAR目标分类方法。为了实现上述目的,本专利技术的技术解决方案包含以下步骤:S1:对MASTAR数据集进行扩充针对现SAR数据训练样本不足的问题,本项操作采用图像翻转、缩放、加噪声、裁剪、移动等方式对现有MASTAR数据集进行扩充来满足训练需求;S2:搭建步进≥2的特殊卷积层传统卷积神经网络采用池化的方式进行下采样,现有的池化方式有最大值池化、最小值池化和平均值池化。本专利技术采用步进≥2的卷积层代替传统的池化层进行下采样,采用该方式既能实现下采样的目的,同时又可以对SAR图像数据进一步特征提取,在提高训练速度的同时,又可以保证整个网络的识别准确度。为了减小步进扩大带来的信息损失,本专利技术同时扩大特殊卷积层核的大小,通过扩大感受野的形式保证信息利用最大化。S3:搭建步进为2的全卷积神经网络(ACN)针对所述步骤S1得到的MASTAR数据集和所述步骤S2所建立的特殊卷积层,本专利技术先构建步进为2的全卷积神经网络。该网络采用步进为2,卷积核大小为2×2的卷积层进行下采样。特征提取的卷积层采用步进为1,卷积核大小为3×3,以此搭建一个全卷积层的网络。该网络在识别准确度上要优于传统的卷进神经网络。S4:搭建步进为3的全卷积神经网络(ACN)针对所述步骤S3搭建的全卷积神经网络,由于采用的步进和传统池化层的步进相当,同时2×2卷积层带来的参数训练使得整个网络的训练速度有所下降。为了提高训练效率,本专利技术扩大了特殊卷积层的步进,采用步进为3的卷积层进行下采样,同时将卷积核的大小调整为3×3,以此来优化训练的效果。S5:搭建步进为4的全卷积神经网络(ACN)为了更进一步说明本专利技术在训练速度上的提升,在所述步骤S4的基础上继续扩大特殊卷积层的步进和卷积核的大小,将其设置为步进为4,大小调整为4×4。该情况下整个网络的训练速度可以提升2倍以上。S6:搭建快速全卷积神经网络(fasterACN)根据步骤S3、步骤S4和步骤S5,综合考虑整个网络的训练速度和识别准确度的情况,本专利技术提出来fasterACN,该网络结合了全卷积神经网络的特点,采用3层卷积层进行特征提取、3层特殊卷积层进行下采样和2层全连接层分类的设计。特征提取的卷积层采用传统的卷积层,步进为1,大小为3×3。下采样的特殊卷积层采用混合递进的方式,分别为步进为4,大小为4×4;步进为3,大小为3×3;步进为2,大小为2×2。这种混合的搭配可以同时发挥全卷积神经网络的速度特点,同时又能保证整个网络的识别准确度不受步进过大带来信息丢失的影响。作为上述方案的进一步改进:所述步骤S1采用15°和17°的MASTAR数据集进行混合训练测试。假设特殊卷积层卷积核大小为n×n,步进为s,输入图片的大小为iw×ih,图片像素填充尺寸为p,卷积层输出图片N的大小为w×h;N是整个卷积过程卷积运算的次数;N=w*h。本专利技术的有益效果:本专利技术针对传统卷积神经网络训练速度慢的问题提出来一种基于快速全卷积神经网络的SAR目标分类方法。通过引入特殊卷积层进行下采样,可以大幅度提高卷积神经网络的训练效率,从而节省大量的训练时间,相比于传统的卷积神经网络,模型训练时间可以减少2.75倍。同时,混合递进的特殊卷积层设置可以保证网络识别准确度不受大步进带来的信息丢失的影响,最终识别准确度高达99.63%。因此,本专利技术可以大大的改善现有SAR技术识别分类方面的困难,提高SAR图像的识别效率和准确度。附图说明通过阅读参照以下附图对非限制性实施例所作的详细描述,本专利技术的其它特征、目的和优点将会变得更明显:图1快速全卷积神经网络(fasterACN)架构示意图。图2是本专利技术针对MASTAR数据集各种数据扩充方式示意图。图3是本专利技术特殊卷积层计算过程示意图。具体实施方式为使本专利技术实现的技术手段、创作特征、达成目的与功效易于明白了解,下面结合具体实施方式,进一步阐述本专利技术。如图1所示,本专利技术提供一种基于快速全卷积神经网络的SAR目标分类方法,包括以下步骤:S1:对MASTAR数据集进行扩充针对现SAR数据训练样本不足的问题,本项操作采用图像翻转、缩放、加噪声、裁剪、移动等方式对现有MASTAR数据集进行扩充来满足训练需求;S2:搭建步进≥2的特殊卷积层传统卷积神经网络采用池化的方式进行下采样,现有的池化方式有最大值池化、最小值池化和平均值池化。本专利技术采用步进≥2的卷积层代替传统的池化层进行下采样,采用该方式既能实现下采样的目的,同时又可以对SAR图像数据进一步特征提取,在提高训练速度的同时,又可以保证整个网络的识别准确度。为了减小步进扩大带来的信息损失,本专利技术同时扩大特殊卷积层核的大小,通过扩大感受野的形式保证信息利用最大化。S3:搭建步进为2的全卷积神经网络(ACN)针对所述步骤S1得到的MASTAR数据集和所述步骤S2所建立的特殊卷积层,本专利技术先构建步进为2的全卷积神经网络。该网络采用步进为2,卷积核大小为2×2的卷积层进行下采样。特征提取的卷积层采用步进为1,卷积核大小为3×3,以此搭建一个全卷积层的网络。该网络在识别准确度上要优于传统的卷进神经网络。S4:搭建步进为3的全卷积神经网络(ACN)针对所述步骤S3搭建的全卷积神经网络,由于采用的步进和传统池化层的步进相当,同时2×2卷积层带来的参本文档来自技高网...

【技术保护点】
1.一种基于快速全卷积神经网络的SAR目标分类方法,其特征在于;包括以下步骤:/nS1:对MASTAR数据集进行扩充/n采用包括图像翻转、缩放、加噪声、裁剪和移动的方式对现有MASTAR数据集进行扩充来满足训练需求;/nS2:搭建步进≥2的特殊卷积层/n采用步进≥2的卷积层进行下采样;同时扩大特殊卷积层核的大小,通过扩大感受野的形式保证信息利用最大化;/nS3:搭建步进为2的全卷积神经网络/n针对所述步骤S1得到的MASTAR数据集和所述步骤S2所建立的特殊卷积层,构建步进为2的全卷积神经网络;卷积核大小为2×2的卷积层进行下采样;特征提取的卷积层采用步进为1,卷积核大小为3×3,以此搭建一个全卷积层的网络;/nS4:搭建步进为3的全卷积神经网络/n针对所述步骤S3搭建的全卷积神经网络,继续采用步进为3的卷积层进行下采样,同时将卷积核的大小调整为3×3;/nS5:搭建步进为4的全卷积神经网络/n在所述步骤S4的基础上继续扩大特殊卷积层的步进和卷积核的大小,采用步进为4,卷积核大小为4×4;/nS6:搭建快速全卷积神经网络/n根据步骤S3、步骤S4和步骤S5,采用3层卷积层进行特征提取、3层特殊卷积层进行下采样和2层全连接层分类的搭建;所述特征提取的卷积层采用步进为1,大小为3×3;下采样的特殊卷积层采用混合递进的方式,分别为步进为4,大小为4×4;步进为3,大小为3×3;步进为2,大小为2×2。/n...

【技术特征摘要】
1.一种基于快速全卷积神经网络的SAR目标分类方法,其特征在于;包括以下步骤:
S1:对MASTAR数据集进行扩充
采用包括图像翻转、缩放、加噪声、裁剪和移动的方式对现有MASTAR数据集进行扩充来满足训练需求;
S2:搭建步进≥2的特殊卷积层
采用步进≥2的卷积层进行下采样;同时扩大特殊卷积层核的大小,通过扩大感受野的形式保证信息利用最大化;
S3:搭建步进为2的全卷积神经网络
针对所述步骤S1得到的MASTAR数据集和所述步骤S2所建立的特殊卷积层,构建步进为2的全卷积神经网络;卷积核大小为2×2的卷积层进行下采样;特征提取的卷积层采用步进为1,卷积核大小为3×3,以此搭建一个全卷积层的网络;
S4:搭建步进为3的全卷积神经网络
针对所述步骤S3搭建的全卷积神经网络,继续采用步进为3的卷积层进行下采样,同时将卷积核的大小调整为3×3;
S5:搭建步进为4的全卷积神经网络
在所述步骤S4的基础上继续扩大特殊卷积层...

【专利技术属性】
技术研发人员:刘聆王芹
申请(专利权)人:长沙一扬电子科技有限公司
类型:发明
国别省市:湖南;43

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1