改进CenterNet的无人机小目标捕获检测方法  

Improved Small Object Detection for UAV Acquisition Based on CenterNet

作　　者：刘鑫 黄进[1] 杨涛 王晴[1] LIU Xin;HUANG Jin;YANG Tao;WANG Qing(School of Electrical Engineering,Southwest Jiaotong University,Chengdu 611756,China)

机构地区：[1]西南交通大学电气工程学院,成都611756

出　　处：《计算机工程与应用》2022年第14期96-104,共9页Computer Engineering and Applications

基　　金：国家自然科学基金(61733015);高铁联合基金(U1934204);四川省重点研发计划(2020YFQ0057);四川省自然资源科研项目(KYL202106-0099)。

摘　　要：小目标因浅层特征语义不足而深层特征信息缺失导致极难检测,而无人机视角场景复杂,检测难度进一步增大。普遍提升小目标检测精度的方法是进行不同层级的特征融合,但这会导致特征高冗余问题,并非所有特征层都值得被激活传递到后方的数据预测中去。针对上述问题对CenterNet进行改造,首次将其与自适应特征激活相结合,提出自适应基础模块(MSA),抑制冗余特征的表达;在主干输出处引入升维全局上下文注意力模块(GC-Block),强化关键点语义信息;用深度可分离卷积与Mish激活搭建高质量解码块(DW),在不增加模型复杂度的情况下提升解码精度。在公开的无人机捕获小目标数据集上进行对比实验,改进算法的AP较原始算法提升了2.2个百分点,召回率提升了2.4个百分点,验证了改进算法对小目标检测任务的有效性。Small targets are extremely difficult to detect due to insufficient low-level semantics and lack of high-level fea-ture information.The scene of the UAV’s perspective is complex,which further increases the difficulty of detection.The general method to improve the detection accuracy of small targets is to perform feature fusion at different levels,but it will cause the problem of high feature redundancy.Not all feature layers are worthy of being activated and passed to the rear data prediction.To solve the above problem,CenterNet is modified,combined with adaptive feature activation for the first time,and modified self-adaptive basic block is proposed to suppress the expression of redundant features;secondly,dimension raising global context block is introduced at the output of backbone to strengthen the key point semantic infor-mation;finally,deep separable convolution and Mish activation are used to build a high-quality decoding block abbreviated as DW,to improve the decoding accuracy without increasing the complexity of the model.Comparing experiments on the public UAV capture small target datasets,the AP of the improved algorithm is 2.2 percentage points higher than the origi-nal algorithm,and the recall is increased by 2.4 percentage points,which verifies the effectiveness of the improved algo-rithm for small target detection tasks.

关 键 词：无人机捕获目标检测 小目标检测 anchor-free 自适应激活 注意力机制 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术]