首页 | 本学科首页   官方微博 | 高级检索  
     

基于迁移集成学习的无人机图像识别算法
引用本文:程千顷,王红军,丁希成,陈璐. 基于迁移集成学习的无人机图像识别算法[J]. 电讯技术, 2023, 63(9): 1277-1284
作者姓名:程千顷  王红军  丁希成  陈璐
作者单位:国防科技大学 电子对抗学院,合肥 230037
基金项目:安徽省融合专项(KY21C008);湖南省研究生科研创新项目(CX20200089);理论研究课题(KY20S011)
摘    要:针对当前小型无人机目标图像识别方法准确率较低的问题,提出了一种基于迁移集成学习的无人机图像识别算法。首先,基于AlexNet、VGGNet-19、Inception-V3以及ResNet-50四种结构具有差异的卷积神经网络对源数据集进行预训练,获取图像的深层次特征;然后,对目标数据集进行迁移学习,得到目标的分类特征,构建分类模型;之后,采用相对多数投票法和加权平均法的集成学习方法,对分类模型进行集成得到迁移集成模型。构建了一个包含小型无人机图像、飞鸟图像以及直升机图像的图像数据集UavNet,在对数据集进行数据增强的基础上开展了图像识别算法性能实验,结果表明,算法对多类目标的识别准确率为99.42%,无人机类目标识别的F1-score指标为99.12%,优于主流的卷积神经网络方法和传统的支持向量机方法,具有一定的理论意义和应用价值。

关 键 词:小型无人机  图像识别  迁移学习  集成学习  深度学习

A UAV Image Recognition Algorithm Based on Transfer Ensemble Learning
CHENG Qianqing,WANG Hongjun,DING Xicheng,CHEN Lu. A UAV Image Recognition Algorithm Based on Transfer Ensemble Learning[J]. Telecommunication Engineering, 2023, 63(9): 1277-1284
Authors:CHENG Qianqing  WANG Hongjun  DING Xicheng  CHEN Lu
Affiliation:College of Electronic Countermeasure,National University of Defense Technology,Hefei 230037,China
Abstract:For the problem of low accuracy of current small unmanned aerial vehicle(UAV) target recognition methods,a UAV recognition algorithm based on transfer learning and ensemble learning is proposed. The algorithm first uses AlexNet,VGGNet-19,Inception-V3 and ResNet-50 convolutional neural networks(CNNs) for pre-training in the source dataset to obtain the deep features of the image. Then transfer learning is used to obtain classification features and generate four classification models on target dataset. A transfer ensemble model is finally established by adopting the relative majority voting method and the weighted average method. In addition,an image dataset called UavNet is built,which includes small UAV images,bird images and helicopter images. The image recognition algorithm experiment is carried out on the UavNet and the result shows that the accuracy of the new algorithm is 99.42%,and the F1-score of UAV target recognition is 99.12%,which is better than those of the mainstream CNNs method and the traditional support vector machines(SVM) methods.
Keywords:small unmanned aerial vehicle  image recognition  transfer learning  ensemble learning  deep learning
点击此处可从《电讯技术》浏览原始摘要信息
点击此处可从《电讯技术》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号