摘要
无人机拍摄的图像具有分辨率高、视野大以及目标小的特点,而现有的目标检测方法对小目标特征的提取能力不足。为此,首先采用以中心点表示目标的检测网络CenterNet,引入可变形双重注意力机制,以提高对小目标的特征表达能力;然后针对原始非极大值抑制难以处理嵌套型冗余框的问题,在冗余检测剔除过程中提出了广义非极大值抑制方法;最后引入LegoNet卷积单元,减少了卷积参数,实现了精度与速度的平衡。实验主要采用的验证数据集为VisDrone2019和UAV_OUC,UAV_OUC数据集相比于VisDrone2019,其图片具有更高的分辨率。相比于CenterNet,所提出的方法在数据集UAV_OUC和VisDrone2019上的检测精度大约分别提高了10%和2%。
Unmanned aerial vehicle(UAV)images have characteristics of high resolution,large field of vision and small target.However,existing object detection methods are generally insufficient in extracting the features of these small targets.Aiming at this problem,a small target detection algorithm is proposed in this paper.First,in order to improve the ability of feature expression for small targets,CenterNet,a detection network which uses center points to represent small targets,is adopted,and a deformable dual attention mechanism is induced.Then on this basis,for the problem of deficiency of original nonmaximum suppression(NMS)in dealing with nested redundant frames,we propose to use a generalized non-maximum suppression(G-NMS)in the process of redundancy detection elimination.Finally,LegoNet convolution unit is introduced to reduce convolution parameters and achieve balance between precision and velocity.The main validation data sets used in this paper are Visdrone 2019 and UAV_OUC.Images in UAV_OUC have higher resolution than those in VisDrone2019.Compared with CenterNet,the detection accuracies of UAV_OUC and VisDrone2019 are improved by about 10%and 2%respectively.
作者
王胜科
任鹏飞
吕昕
庄新发
WANG Shengke;REN Pengfei;Lü Xin;ZHUANG Xinfa(College of Information Science and Engineering,Ocean University of China,Qingdao 266100,Shandong,China)
出处
《应用科学学报》
CAS
CSCD
北大核心
2021年第4期650-659,共10页
Journal of Applied Sciences
基金
国家自然科学基金(No.41927805,No.U17062189,No.61602229,No.41606198,No.61501417,No.41706010)
国家重点研发计划基金(No.2018YFB1701802)
装备预研教育部联合基金(No.6141A020337)
山东省自然科学基金(No.ZR2016FM13,No.ZR2016FB02)资助。
关键词
无人机
高分辨率
小目标检测
中心点检测
注意力机制
unmanned aerial vehicle(UAV)
high resolution
small target detection
center point detection
attention mechanism