买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:西安电子科技大学
摘要:本发明提供了一种有限样本下属性引导的SAR图像生成方法。主要解决训练样本有限的情况下,现有的SAR图像生成对抗网络容易产生网络崩塌或者生成的仿真图像的真实感较差的问题。本发明的步骤如下:1构建训练集;2构建生成器网络;3构建判别器网络;4构建属性引导的生成对抗网络;5训练属性引导的生成对抗网络;6使用训练好的属性引导的生成对抗网络生成SAR仿真图像。本发明可在训练样本数量有限情况下训练属性引导的生成对抗网络,再利用训练好的属性引导的生成对抗网络,生成指定类别和方位角度的SAR图像,且生成的SAR图像的真实性更高。
主权项:1.一种有限样本下属性引导的SAR图像生成方法,其特征在于,构建一个由属性引导的生成对抗网络,在训练集中目标域数据集的样本数量有限的情况下训练属性引导的生成对抗网网络,该方法的具体步骤如下:1构建训练集:1a提取包含M个类别的带有目标方位角度信息的SAR图像组成目标域数据集,每个类别的SAR图像数量均为K;提取包含N个类别的带有目标方位角度信息的SAR图像组成源域数据集,将目标域数据集和源域数据集组成SAR数据集,其中,源域数据集每个类别的SAR图像数量均大于等于200;N≥3,1≤M≤3,K≤30;1b对SAR数据集中每个类别目标的SAR图像设置类别标签;1c将SAR数据集中每幅SAR图像的大小裁剪为64×64,对裁剪后的每幅SAR图像进行L2归一化处理,得到归一化后的SAR数据集;1d将归一化后的SAR数据集中每幅SAR图像的像素值范围转化成0到255之间后得到训练集;2构建生成器网络:2a搭建一个21层的生成器网络,其结构依次为:第1叠加层,全连接层,第1激活函数层,reshape层,第2叠加层,第1反卷积层,归一化层,第2激活函数层,第3叠加层,第2反卷积层,归一化层,第3激活函数层,第4叠加层,第3反卷积层,归一化层,第4激活函数层,第5叠加层,第4反卷积层,归一化层,第5激活函数层,输出层;所述归一化层采用BatchNormalization函数实现;所述第1至第4激活函数层均采用ReLU函数实现,第5激活函数层采用Tanh函数实现;2b设置生成器网络的每层参数:将第1至第4反卷积层的卷积核大小均设置为5×5,卷积核的个数分别设置为128、64、32、16,卷积步长均设置为2;将全连接层的神经元个数设置为2048;3构建判别器网络:3a搭建一个16层的判别器网络,其结构依次为:第1叠加层,第1卷积层,激活函数层,第2叠加层,第2卷积层,归一化层,激活函数层,第3叠加层,第3卷积层,归一化层,激活函数层,第4叠加层,第4卷积层,归一化层,激活函数层,并联的三个全连接层;所述激活函数层均采用LeakyReLU函数实现;3b设置判别器网络的每层参数:将第1至第4卷积层的卷积核大小均设置为5×5,卷积核的个数分别设置为16、32、64、128,卷积步长均设置为2;将每个激活函数层的LeakyReLU函数的斜率均设置为0.2;将三个并联的全连接层的神经元个数分别设置为1、1、C,其中C的取值与目标类别总数M+N相等;4将生成器网络和判别器网络级联成属性引导的生成对抗网络;5训练属性引导的生成对抗网络:5a从训练集的全部类别中随机选取的n个类别,n≤3,每个类别选取m幅SAR图像组成训练集的支撑集,20≤m≤23;5b将支撑集中所有的SAR图像、支撑集中每张SAR图像对应的类别标签、支撑集中每张SAR图像对应的方位角度信息同时输入到判别器网络中,判别器网络对输入的每张SAR图像评估后输出预测为真实图像的概率值、类别预测概率值和方位角度预测值,并利用判别器网络的损失函数计算判别器网络的损失值;5c使用梯度下降法,计算每个卷积层的每个卷积核的梯度、全连接层的梯度,使用学习率为0.00005的Adam优化器更新判别器网络每个卷积层的每个卷积核的权重、全连接层的权重;5d生成B个满足正态分布的1×100维的随机向量,B=n×m;5e将随机向量、支撑集中所有SAR图像对应的类别标签、支撑集中所有SAR图像对应的方位角度信息分别输入到生成器网络中,输出F幅SAR生成图像,F=n×m;5f将F幅SAR生成图像、支撑集中每个SAR图像对应的类别标签、支撑集中每个SAR图像对应的方位角度信息分别输入到判别器网络中,判别器网络对输入的SAR生成图像评估后分别输出预测为真实图像的概率值、类别预测概率值和方位角度预测值,并利用生成器网络的损失函数计算生成器网络的损失值;5g利用生成器网络的损失值和梯度下降法,计算生成器网络每个反卷积层的每个卷积核的梯度,使用学习率为0.00005的Adam优化器更新生成器网络每个反卷积层的每个卷积核的权重;5h判断判别器网络的损失函数和生成器网络的损失函数是否均收敛,若是,则得到训练好的属性引导的生成对抗网络,否则,执行步骤5a;6生成SAR图像:生成H个满足正态分布的1×100维的随机向量,将随机向量、P个类别标签、Q个方位角度信息输入到训练好的属性引导的生成对抗网络,输出与类别标签和方位角度信息对应的SAR生成图像,其中,H、P、Q的取值与待生成的SAR图像的数量相等。
全文数据:
权利要求:
百度查询: 西安电子科技大学 有限样本下属性引导的SAR图像生成方法
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。