循环神经网络多标签航空图像分类

• 作者: 张叶 陈科峻
• 来源期刊: 光学精密工程
• 学科: 工学
• 关键词: 航空图像分类 多标签 注意力机制 多尺度 卷积神经网络 长短期记忆网络
• 年，卷（期）: 2020,（6）
• 页码范围: 1404-1413
• 页数: 10页
• 分类号: TP391.7
• 字数: 5102字
• 语种: 中文
• DOI: 10.3788/OPE.20202806.1404

摘要

由于航空图像背景复杂,包含的物体类别多样,航空图像分类任务仍然面临困难.针对传统航空图像多标签分类算法准确率低、泛化性差的问题,本文提出了一种基于循环神经网络多标签航空图像分类方法.首先,采用超像素分割获取图像的低层特征,通过注意力机制生成注意力特征图;接着,采用交叉验证的方式获取最佳的图像尺度,将多尺度注意力特征图嵌入卷积神经网络中对图像进行特征提取;最后,采用改进的双向长短期记忆网络挖掘标签之间的相关性,改进的双向长短期记忆网络增加了输入门到输出门之间的连接,使输入状态可以更好地控制每一内存单元输出的信息,并且将遗忘门和输入门合并成单一的更新门,使得改进的双向长短期记忆网络可以学到更长时期的历史信息.结果 显示,在图像变换尺度为1,1.3,2时,模型在UCM多标签数据集上的精确率和召回率分别达到了85.33％和87.05％,F1值达到了0.862.本文方法相比于原始VGGNet16模型,精确率提高了7.25％,召回率提高了8.94％.实验表明,该方法可以有效提高航空图像多标签分类任务的准确率.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	张叶	中国科学院大学长春光学精密机械与物理研究所应用光学国家重点实验室	36	621	12.0	24.0
2	陈科峻	中国科学院大学长春光学精密机械与物理研究所应用光学国家重点实验室	2	0	0.0	0.0