摘要
空中无人机目标识别是现代军事、航空领域的迫切需求,由于目前无人机的功能和种类繁多,对于新机型很难采集大量的无人机样本用于训练目标识别模型;针对该问题,提出了一种基于模型微调的空中无人机小样本目标识别方法;方法以Faster R-CNN为基础架构,首先采用具有大量标记样本的常见机型数据预训练Faster R-CNN模型;然后将基础架构最后的分类层替换为余弦度量,构建联合新机型与常见机型的小样本平衡数据集以较小的学习率微调分类层;实验结果表明,在标记样本数量为5、10和50的情况下,基于模型微调的小样本目标识别模型的mAP分别为88.6%,89.2%和90.8%,能够满足空中无人机小样本目标识别任务需求,且优于其它小样本目标识别方法。
Aerial drone object recognition is an urgent demand in modern military and aviation fields.Due to the various functions and types of drones at present,it is difficult to collect a large number of new drone samples for training object recognition models.In order to solve this problem,a few-shot object recognition method for aerial drones based on model fine-tuning is proposed.The method is based on a Faster R-CNN architecture.Firstly,the Faster R-CNN model is pretrained by using the data of common drones with a large number of labeled samples.Then,the last classification layer of the infrastructure is replaced by the cosine measurement,and the small balanced datasets of new drones and common drones are constructed and jointed to fine-tune the classification layer with a small learning rate.Experimental results show that with the number of labeled samples of 5,10 and 50,the mAP of few-shot object recognition model based on the model fine-tuning is 88.6%,89.2%and 90.8%respectively,which can meet the requirements of few-shot target recognition task of aerial drones,and the proposed method is better than other few-shot target recognition methods.
作者
黄灿
HUANG Can(91550 Army,Dalian116023,China)
出处
《计算机测量与控制》
2024年第1期268-274,共7页
Computer Measurement &Control
