注意力机制与卷积神经网络融合的军队标号识别方法  

Recognition Method for Military Symbology Based on Fusion of Attention Mechanism and Convolutional Neural Network

作　　者：杨云航 闵连权[1] 侯翔 王佳伟 YANG Yunhang;MIN Lianquan;HOU Xiang;WANG Jiawei(Information Engineering University, Zhengzhou 450001, China;Aerospace Engineering University, Beijing 101416, China)

机构地区：[1]信息工程大学,河南郑州450001 [2]航天工程大学,北京101416

出　　处：《测绘科学技术学报》2021年第2期200-205,212,共7页Journal of Geomatics Science and Technology

基　　金：国家自然科学基金项目(41471337)。

摘　　要：手绘军队标号识别是基于草图的战场态势表达的重要组成部分。针对现有的草图特征手工提取方法费时费力,以及依靠数据驱动的深度学习方法泛化能力受到训练数据多样化制约的问题,提出一种注意力机制与卷积神经网络融合的军队标号识别方法。首先构建了特征提取模型Sketch-Net,实现手绘军队标号特征的初步提取;然后引入注意力机制模块,分别捕获特征在通道和空间位置中的依赖关系,增强模型提取特征信息的有效性。结果表明,提出的方法在建立的军队标号数据集上的识别正确率达到95.75%,能有效应用于手绘军队标号的识别。Sketch recogition for military symbology is a significant component of sketch-based military battlefield situation expression.Existing sketch feature extraction methods are time-consuming and laborious,and the generalization ability of data-driven deep learning methods is restricted by the diversity of training data.A sketch recognition method based on the fusion of attention mechanism and convolution neural network is proposed in the paper.Firstly,the preliminary features of sketch military symbology are extracted by the Sketch-Net model.Then,the attention mechanism module is introduced to capture the dependency of features in channel and spatial location respectively,to enhance the validity of extracted feature information of the model.The experimental results show that the accuracy of the proposed method in the SF-Sketch dataset can reach 95.75%,which proves that the method can be effectively applied to sketch recognition of military symbology.

关 键 词：注意力机制 卷积神经网络 军队标号 草图识别 特征提取 

分 类 号：P289[天文地球—地图制图学与地理信息工程]