论文摘要
针对SAR图像舰船目标识别中存在的数据不平衡问题提出了一种基于批内平衡采样和模型微调的两阶段迁移学习方法。首先使用批内平衡采样方法得到类间数量平衡的训练集,然后使用该数据集预训练模型,最后通过迁移学习和模型微调继续训练不平衡数据并完成测试。针对一般三通道CNN模型在处理单通道SAR图像时会出现参数冗余的问题,设计了一种用于SAR图像识别的轻量化CNN模型。通过单通道卷积核、深度可分离卷积和用全局平均池化代替全连接层3种策略有效降低了模型的参数量。在公开数据集OpenSARShip上的实验结果表明:所提方法有效提升了少数类的识别精度,缓解了数据不平衡问题对识别结果的影响;所提轻量化CNN模型在保证识别精度基本不变的前提下,使传统三通道CNN模型的模型大小和单次迭代时间分别降低约58.86%和63.62%。
论文目录
文章来源
类型: 期刊论文
作者: 邵嘉琦,曲长文,李健伟,彭书娟
关键词: 舰船识别,卷积神经网络,数据不平衡,轻量化模型
来源: 电光与控制 2019年09期
年度: 2019
分类: 工程科技Ⅱ辑,信息科技
专业: 武器工业与军事技术,船舶工业,电信技术,计算机软件及计算机应用,自动化技术
单位: 海军航空大学
分类号: U675.79;E91;TN957.52;TP183
页码: 90-97
总页数: 8
文件大小: 3008K
下载量: 386