CN114943251B - 一种基于融合注意力机制的无人机目标识别方法 - Google Patents
一种基于融合注意力机制的无人机目标识别方法 Download PDFInfo
- Publication number
- CN114943251B CN114943251B CN202210548931.XA CN202210548931A CN114943251B CN 114943251 B CN114943251 B CN 114943251B CN 202210548931 A CN202210548931 A CN 202210548931A CN 114943251 B CN114943251 B CN 114943251B
- Authority
- CN
- China
- Prior art keywords
- unmanned aerial
- aerial vehicle
- layer
- module
- attention
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Active
Links
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/047—Probabilistic or stochastic networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F2218/00—Aspects of pattern recognition specially adapted for signal processing
- G06F2218/08—Feature extraction
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F2218/00—Aspects of pattern recognition specially adapted for signal processing
- G06F2218/12—Classification; Matching
-
- Y—GENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
- Y02—TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
- Y02T—CLIMATE CHANGE MITIGATION TECHNOLOGIES RELATED TO TRANSPORTATION
- Y02T10/00—Road transport of goods or passengers
- Y02T10/10—Internal combustion engine [ICE] based vehicles
- Y02T10/40—Engine management systems
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- Evolutionary Computation (AREA)
- Molecular Biology (AREA)
- Artificial Intelligence (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Data Mining & Analysis (AREA)
- Health & Medical Sciences (AREA)
- General Health & Medical Sciences (AREA)
- Life Sciences & Earth Sciences (AREA)
- Computing Systems (AREA)
- General Engineering & Computer Science (AREA)
- General Physics & Mathematics (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Probability & Statistics with Applications (AREA)
- Image Analysis (AREA)
Abstract
本发明属于无人机识别技术领域,具体涉及一种基于融合注意力机制的无人机目标识别方法。本发明的识别模型由数据输入层、预处理层、卷积模块、注意力模块、全连接层以及softmax分类层组成,数据输入层与预处理层用于处理输入的无人机微多普勒谱图;由卷积模块与注意力模块提取谱图深度特征;采用全连接层与分类层实现对无人机目标的分类识别。由于注意力模块是针对微多普勒谱图的特点设计的,能够提取有利于分类的局部特征信息,注意力模块与卷积模块相融合可以提取到分类性能更好的特征,从而进一步改善对目标的识别性能。仿真实验验证了本文方法的有效性。
Description
技术领域
本发明属于无人机识别技术领域,具体涉及一种基于融合注意力机制的无人机目标识别方法。
背景技术
随着科技的发展,不论是在军事领域还是在民用领域,无人机都得到了广泛的应用,但同时也带来了撞机、恐怖袭击、黑飞等安全问题。因此,能快速且准确地识别出无人机类型以便及时做出应对变得尤为重要。
目前,无人机目标识别的主要方法是利用无人机旋翼的微动生成微多普勒谱图,然后基于谱图进行分类识别。传统的无人机目标识别方法需要人为设计要提取的识别特征,这不仅耗时而且准确率不高。近年来,基于卷积神经网络的识别方法在图像领域取得巨大成功,许多学者已经将其引入到无人机目标识别领域,但卷积神经网络关注的是全部区域的特征信息,往往会忽视一些有于分类的局部区域,而注意力机制正好是关注一些有区分力的局部区域,因此融合注意力机制的卷积神经网络可以进一步提高无人机目标识别的准确率。
发明内容
本发明将注意力机制与卷积神经网络相融合,提出了一种基于融合注意力机制的无人机目标识别方法。
本发明的技术方案为:
一种基于融合注意力机制的无人机目标识别方法,包括以下步骤:
S1、获取无人机的微多普勒谱图构成训练数据集;
S2、构建基于融合注意力机制的无人机目标识别模型,包括输入层、第一卷积模块、注意力模块、第二卷积模块、第三卷积模块、全连接层和分类层;所述输入层接收训练数据并进行归一化后输入第一卷积模块,第一卷积模块输出特征图x到注意力模块,注意力模块采用的注意力函数为:
其中,h为x的高度,w为x的宽度,x(i,j)是特征图x中第i行第j列的像素值,r为预先设定参数,表示被识别关注的特征所占的高度比,根据实际需要由实验确定,经过注意力函数g(·)后将生成一个与特征图x同高同宽的二维掩膜,令被识别关注的区域中的像素值为1,而其他区域的像素值为0,经过注意力机制处理后的特征表达为:
f(g(x(i,j)),x(i,j))=g(x(i,j))×x(i,j),i∈[0,h),j∈[0,w)
再经过第二卷积模块和第三卷积模块进行特征提取,其中第二卷积模块和第三卷积模块是常规的卷积模块,包含多个卷积核,卷积核的大小与数量由实验确定,第三卷积模块的输出即是提取的二维深度特征,后面串接一个全连接层,全连接层将提取的二维深度特征转化为一维向量特征,最后采用softmax分类器实现分类层,全连接层输出的一维向量特征作为分类层的输入,以分类层输出向量中最大的值对应的标签作为该目标的类别标签;
S3、采用S1中训练数据集对S2中无人机目标识别模型进行训练后得到训练好的无人机识别模型;
S4、获取待识别的无人机微多普勒谱图,输入采用训练好的无人机识别模型进行识别。
本发明的有益效果为,由于注意力模块是针对微多普勒谱图的特点设计的,能够提取有利于分类的局部特征信息,注意力模块与卷积模块相融合可以提取到分类性能更好的特征,从而进一步改善对目标的识别性能。仿真实验验证了本文方法的有效性。。
附图说明
图1为基于融合注意力机制的无人机目标识别模型结构示意图。
图2为二旋翼与四旋翼无人机的微多普勒谱图示意图。
具体实施方式
下面结合附图和仿真对本发明进行详细的说明:
如图1所示,本发明提出的基于融合注意力机制的无人机目标识别模型由数据输入层、预处理层、卷积模块、注意力模块、全连接层以及softmax分类层组成,其中,卷积模块与注意力模块用于提取特征,由全连接层与分类层实现对目标的分类识别。输入部分由数据集输入层与预处理层组成。其中,输入的数据集是由无人机的微多普勒谱图组成的训练集或者测试集;预处理层将对输入的谱图进行归一化处理。
特征提取部分由卷积模块1、注意力模块、卷积模块2以及卷积模块3组成。其中,卷积模块即为一般的卷积网络,不同卷积模块的区别主要在于卷积核的大小不一样;注意力模块的实现如下:
如图2所示,(a)是二旋翼无人机的微多普勒谱图,图(b)是四旋翼无人机的微多普勒谱图,从图2可见,其关键的区分特征位于图像的中间区域,因此,希望网络可以更多的关注特征图的中间区域,基于此特点设计了一种注意力机制。设特征图x的高度为h,宽度为w,主要特征所占的高度比为r,则设计注意力函数g(x(i,j))如下:
其中,x(i,j)是特征图x中第i行第j列的像素值,即经过g(·)后将生成一个与特征图x同高同宽的二维掩膜,其主要被关注区域中的像素值为1,而其他区域的像素值为0。经过注意力机制处理后的特征表达为:
f(g(x(i,j)),x(i,j))=g(x(i,j))×x(i,j),i∈[0,h),j∈[0,w)
即经过注意力机制处理后的特征仅关注有区分力的局部区域特征,而忽视了其他区域的特征。因此,解决了常规卷积网络会关注整个图像区域特征信息的问题。
分类识别部分由全连接层与分类层组成。全连接层将提取的二维深度特征转化为一维向量特征,以便之后的分类识别;采用softmax分类器实现分类层,以softmax分类器输出向量中最大的值对应的标签作为该目标的类别标签。
仿真示例
通过仿真实验设计了6种种类的无人机,分别是单旋翼无人机、双旋翼无人机、三旋翼无人机、四旋翼无人机、六旋翼无人机和八旋翼无人机。其中,无人机的旋翼到其中心距离为0.8m,桨叶长度为0.3m,旋翼转速为30r/s;仿真雷达载频为34.6GHz(Ka波段),脉冲重复频率为125KHz;无人机中心与雷达的距离为100m,无人机相对于雷达的俯仰角为10°、方位角为45°,信噪比为15dB。
每类目标的总观测时间为15s,取每0.05s的雷达回波信号作为一帧,每帧之间的重复率为0.6,则每类目标均可得到15/(0.05*(1-0.6))=750个雷达回波信号,随机选取其中的600个雷达回波信号作为样本。
对获取的雷达回波数据进行短时傅里叶变换(STFT)得到微多普勒谱图,则每类目标均有600张微多普勒谱图,每类样本按照7:3的比例划分训练集与测试集,每类目标有420个训练样本、180个测试样本,则6类目标总共有2520个训练样本和1080个测试样本。
利用常规CNN方法和本文方法对以上数据集进行识别实验,结果如表1所示。其中,学习率为0.001,迭代次数为200次,采用交叉熵损失函数,卷积模块1含有32个5*5的卷积核,卷积模块2含有64个5*5的卷积核,卷积模块3含有32个3*3的卷积核;注意力模块中的高度比r取0.7。
表1两种网络模型的识别结果
由表1可知CNN+注意力机制方法的平均识别准确率达到了97.68%,且直接使用CNN的方法与CNN+注意力机制的方法所需的识别时间差不多,但是CNN+注意力机制方法的平均识别准确率要比常规CNN方法高出了0.55%,以上结果表明本文提出的方法是有效的。
Claims (1)
1.一种基于融合注意力机制的无人机目标识别方法,其特征在于,包括以下步骤:
S1、获取无人机的微多普勒谱图构成训练数据集;
S2、构建基于融合注意力机制的无人机目标识别模型,包括输入层、第一卷积模块、注意力模块、第二卷积模块、第三卷积模块、全连接层和分类层;所述输入层接收训练数据并进行归一化后输入第一卷积模块,第一卷积模块输出特征图x到注意力模块,注意力模块采用的注意力函数为:
其中,h为x的高度,w为x的宽度,x(i,j)是特征图x中第i行第j列的像素值,r为预先设定参数,表示被识别关注的特征所占的高度比,经过注意力函数g(·)后将生成一个与特征图x同高同宽的二维掩膜,令被识别关注的区域中的像素值为1,而其他区域的像素值为0,经过注意力机制处理后的特征表达为:
f(g(x(i,j)),x(i,j))=g(x(i,j))×x(i,j),i∈[0,h),j∈[0,w)
再经过第二卷积模块和第三卷积模块进行特征提取,第三卷积模块的输出即是提取的二维深度特征,后面串接一个全连接层,全连接层将提取的二维深度特征转化为一维向量特征,最后采用softmax分类器实现分类层,全连接层输出的一维向量特征作为分类层的输入,以分类层输出向量中最大的值对应的标签作为该目标的类别标签;
S3、采用S1中训练数据集对S2中无人机目标识别模型进行训练后得到训练好的无人机识别模型;
S4、获取待识别的无人机微多普勒谱图,输入采用训练好的无人机识别模型进行识别。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210548931.XA CN114943251B (zh) | 2022-05-20 | 2022-05-20 | 一种基于融合注意力机制的无人机目标识别方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210548931.XA CN114943251B (zh) | 2022-05-20 | 2022-05-20 | 一种基于融合注意力机制的无人机目标识别方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN114943251A CN114943251A (zh) | 2022-08-26 |
CN114943251B true CN114943251B (zh) | 2023-05-02 |
Family
ID=82909087
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202210548931.XA Active CN114943251B (zh) | 2022-05-20 | 2022-05-20 | 一种基于融合注意力机制的无人机目标识别方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN114943251B (zh) |
Citations (10)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109543606A (zh) * | 2018-11-22 | 2019-03-29 | 中山大学 | 一种加入注意力机制的人脸识别方法 |
CN110516596A (zh) * | 2019-08-27 | 2019-11-29 | 西安电子科技大学 | 基于Octave卷积的空谱注意力高光谱图像分类方法 |
CN111160527A (zh) * | 2019-12-27 | 2020-05-15 | 歌尔股份有限公司 | 一种基于mask rcnn网络模型的目标识别方法、装置 |
CN112200161A (zh) * | 2020-12-03 | 2021-01-08 | 北京电信易通信息技术股份有限公司 | 一种基于混合注意力机制的人脸识别检测方法 |
CN112464792A (zh) * | 2020-11-25 | 2021-03-09 | 北京航空航天大学 | 一种基于动态卷积的遥感图像舰船目标细粒度分类方法 |
CN112966555A (zh) * | 2021-02-02 | 2021-06-15 | 武汉大学 | 一种基于深度学习和部件先验的遥感影像飞机识别方法 |
CN113191185A (zh) * | 2021-03-10 | 2021-07-30 | 中国民航大学 | 一种利用Dense2Net的雷达探测无人机目标分类方法 |
CN114140683A (zh) * | 2020-08-12 | 2022-03-04 | 天津大学 | 一种航拍图像目标检测的方法、设备与介质 |
CN114445366A (zh) * | 2022-01-26 | 2022-05-06 | 沈阳派得林科技有限责任公司 | 基于自注意力网络的长输管道射线影像缺陷智能识别方法 |
CN114488140A (zh) * | 2022-01-24 | 2022-05-13 | 电子科技大学 | 一种基于深度迁移学习的小样本雷达一维像目标识别方法 |
Family Cites Families (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US11915602B2 (en) * | 2020-07-31 | 2024-02-27 | North Carolina State University | Drone detection, classification, tracking, and threat evaluation system employing field and remote identification (ID) information |
-
2022
- 2022-05-20 CN CN202210548931.XA patent/CN114943251B/zh active Active
Patent Citations (10)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109543606A (zh) * | 2018-11-22 | 2019-03-29 | 中山大学 | 一种加入注意力机制的人脸识别方法 |
CN110516596A (zh) * | 2019-08-27 | 2019-11-29 | 西安电子科技大学 | 基于Octave卷积的空谱注意力高光谱图像分类方法 |
CN111160527A (zh) * | 2019-12-27 | 2020-05-15 | 歌尔股份有限公司 | 一种基于mask rcnn网络模型的目标识别方法、装置 |
CN114140683A (zh) * | 2020-08-12 | 2022-03-04 | 天津大学 | 一种航拍图像目标检测的方法、设备与介质 |
CN112464792A (zh) * | 2020-11-25 | 2021-03-09 | 北京航空航天大学 | 一种基于动态卷积的遥感图像舰船目标细粒度分类方法 |
CN112200161A (zh) * | 2020-12-03 | 2021-01-08 | 北京电信易通信息技术股份有限公司 | 一种基于混合注意力机制的人脸识别检测方法 |
CN112966555A (zh) * | 2021-02-02 | 2021-06-15 | 武汉大学 | 一种基于深度学习和部件先验的遥感影像飞机识别方法 |
CN113191185A (zh) * | 2021-03-10 | 2021-07-30 | 中国民航大学 | 一种利用Dense2Net的雷达探测无人机目标分类方法 |
CN114488140A (zh) * | 2022-01-24 | 2022-05-13 | 电子科技大学 | 一种基于深度迁移学习的小样本雷达一维像目标识别方法 |
CN114445366A (zh) * | 2022-01-26 | 2022-05-06 | 沈阳派得林科技有限责任公司 | 基于自注意力网络的长输管道射线影像缺陷智能识别方法 |
Non-Patent Citations (4)
Title |
---|
Payal Mittal等.Deep learning-based object detection in low-altitude UAV datasets: A survey.Image and Vision Computing.2020,第104卷1-24. * |
Wei Sun等 .RSOD: Real-time small object detection algorithm in UAV-based traffic monitoring.Applied Intelligence .2021,第52卷8448–8463. * |
杨威.基于机载视觉的无人机自主目标检测技术研究.中国优秀硕士学位论文全文数据库 工程科技Ⅱ辑.2022,C031-457. * |
莫文昊.基于深度学习的航拍图像目标检测算法.中国优秀硕士学位论文全文数据库 工程科技Ⅱ辑.2021,C031-557. * |
Also Published As
Publication number | Publication date |
---|---|
CN114943251A (zh) | 2022-08-26 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN108388927B (zh) | 基于深度卷积孪生网络的小样本极化sar地物分类方法 | |
CN108921030B (zh) | 一种sar自动目标识别方法 | |
CN110197209B (zh) | 一种基于多特征融合的辐射源识别方法 | |
CN107830996B (zh) | 一种飞行器舵面***故障诊断方法 | |
CN107103338A (zh) | 融合卷积特征和集成超限学习机的sar目标识别方法 | |
CN111352086B (zh) | 一种基于深度卷积神经网络的未知目标识别方法 | |
CN104237883A (zh) | 一种采用稀疏表示的机载雷达空时自适应处理方法 | |
CN112446874A (zh) | 一种人机协作自主等级毁伤评估方法 | |
CN110136162B (zh) | 无人机视角遥感目标跟踪方法及装置 | |
CN110232371B (zh) | 基于小样本的高精度hrrp雷达多目标识别方法 | |
CN113191185A (zh) | 一种利用Dense2Net的雷达探测无人机目标分类方法 | |
CN104732224B (zh) | 基于二维泽尔尼克矩特征稀疏表示的sar目标识别方法 | |
CN113269203B (zh) | 一种用于多旋翼无人机识别的子空间特征提取方法 | |
CN114943251B (zh) | 一种基于融合注意力机制的无人机目标识别方法 | |
CN114821335B (zh) | 基于深度特征与线性判别特征融合的未知目标判别方法 | |
CN116682015A (zh) | 一种基于特征解耦的跨域小样本雷达一维像目标识别方法 | |
CN108106500B (zh) | 一种基于多传感器的导弹目标类型识别方法 | |
CN115656958A (zh) | 实时航迹起始和航迹分类的检测方法及检测装置 | |
CN115902804A (zh) | 一种无人机集群类型识别方法和*** | |
CN115909086A (zh) | 基于多级增强网络的sar目标检测识别方法 | |
CN114137518A (zh) | 一种雷达高分辨距离像开集识别方法及装置 | |
He et al. | A multi-scale radar HRRP target recognition method based on pyramid depthwise separable convolution network | |
CN114937150B (zh) | 一种基于深度阈值残差网络的无人机目标识别方法 | |
CN112257792A (zh) | 一种基于svm的实时视频目标动态分类方法 | |
CN114936576B (zh) | 多旋翼无人机识别中的子块差分编码分布特征提取方法 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |