[发明专利]一种基于深度学习的可解释SAR图像的识别方法在审
申请号: | 202210909719.1 | 申请日: | 2022-07-29 |
公开(公告)号: | CN115147725A | 公开(公告)日: | 2022-10-04 |
发明(设计)人: | 李鹏;胡晓伟;冯存前;汪洋;齐铖 | 申请(专利权)人: | 中国人民解放军空军工程大学 |
主分类号: | G06V20/10 | 分类号: | G06V20/10;G06V10/764;G06V10/774;G06V10/82;G06N3/04;G06N3/08;G01S7/41 |
代理公司: | 北京沃知思真知识产权代理有限公司 11942 | 代理人: | 孙建 |
地址: | 710051 陕西*** | 国省代码: | 陕西;61 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 一种 基于 深度 学习 可解释 sar 图像 识别 方法 | ||
1.一种基于深度学习的可解释SAR图像的识别方法,其特征在于,包括以下步骤:
S1:在网络在ResNet-18框架的基础上,构建网络模型;
S2:根据网络架构,对模型的感受野、步长和填充进行设置;
S3:通过模型追踪图像中每块区域,并设置得分规则;
S4:确定模型预测的类别;
S5:基于SAR图像的MSTAR数据集对模型进行训练和测试,直至完成模型训练;
S6:通过训练好的模型对图像进行识别。
2.根据权利要求1所述的基于深度学习的可解释SAR图像的识别方法,其特征在于,所述模型的感受野小于图像的尺寸3-7倍。
3.根据权利要求2所述的基于深度学习的可解释SAR图像的识别方法,其特征在于,所述模型的每一层卷积层的步长为1。
4.根据权利要求3所述的基于深度学习的可解释SAR图像的识别方法,其特征在于,在步骤S1中,在ResNet-18框架的基础上进行了改进,将原有的卷积核替换为1×1和3×3个卷积核,每个卷积层后面有一个正则化层和ReLU层。
5.根据权利要求4所述的基于深度学习的可解释SAR图像的识别方法,其特征在于,用于步骤S2中,网络卷积层数能够进行调节。
6.根据权利要求5所述的基于深度学习的可解释SAR图像的识别方法,其特征在于,在步骤S3中,得分规则为热图上的每个值对应的图像区域代表了此区域对决策结果的贡献大小,将得到的热图中每一个像素值求和就可得到每一类的得分。
7.根据权利要求6所述的基于深度学习的可解释SAR图像的识别方法,其特征在于,在步骤S4中,将每一类得分经过softmax分类器得到每一类的概率,取最大概率的一类为模型预测的类别。
8.根据权利要求7所述的基于深度学习的可解释SAR图像的识别方法,其特征在于,在步骤S2中,在有n个卷积层的深度卷积网络的中间没有池化层时,最后一层卷积层特征图中每个单元对应的输入图像上的感受野大小的计算公式为:
其中,RFn为待计算的第n层接收野的大小,RFn-1为计算得到的第n-1层接收野的大小,fn为第n个卷积核的大小,si为第i层卷积对应的步长。
9.根据权利要求1-8任一所述的基于深度学习的可解释SAR图像的识别方法,其特征在于,模型的训练和测试具体包括以下步骤:
S501:选择Adam优化器为模型的优化器,设置学习率为0.001,损失函数为交叉熵损失;
S502:将MSTAR数据集中带有类别标签的训练数据以合适的批次大小输入到模型中,使用模型的输出和真实标签计算交叉熵损失;
S503:以损失的下降方向进行梯度反向传播更新模型的各层参数;
S504:重复S502、S503步骤,达到预设的迭代步骤后停止,模型即训练完毕。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于中国人民解放军空军工程大学,未经中国人民解放军空军工程大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202210909719.1/1.html,转载请声明来源钻瓜专利网。