Computer Engineering and Applications ›› 2022, Vol. 58 ›› Issue (14): 96-104.DOI: 10.3778/j.issn.1002-8331.2111-0033

• Pattern Recognition and Artificial Intelligence • Previous Articles     Next Articles

Improved Small Object Detection for UAV Acquisition Based on CenterNet

LIU Xin, HUANG Jin, YANG Tao, WANG Qing   

  1. School of Electrical Engineering, Southwest Jiaotong University, Chengdu 611756, China
  • Online:2022-07-15 Published:2022-07-15

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

刘鑫,黄进,杨涛,王晴   

  1. 西南交通大学 电气工程学院,成都 611756

Abstract: Small targets are extremely difficult to detect due to insufficient low-level semantics and lack of high-level feature 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 information;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 original algorithm, and the recall is increased by 2.4 percentage points, which verifies the effectiveness of the improved algorithm for small target detection tasks.

Key words: UAV capture target detection, small object detection, anchor-free, adaptive activation, attention mechanism

摘要: 小目标因浅层特征语义不足而深层特征信息缺失导致极难检测,而无人机视角场景复杂,检测难度进一步增大。普遍提升小目标检测精度的方法是进行不同层级的特征融合,但这会导致特征高冗余问题,并非所有特征层都值得被激活传递到后方的数据预测中去。针对上述问题对CenterNet进行改造,首次将其与自适应特征激活相结合,提出自适应基础模块(MSA),抑制冗余特征的表达;在主干输出处引入升维全局上下文注意力模块(GC-Block),强化关键点语义信息;用深度可分离卷积与Mish激活搭建高质量解码块(DW),在不增加模型复杂度的情况下提升解码精度。在公开的无人机捕获小目标数据集上进行对比实验,改进算法的AP较原始算法提升了2.2个百分点,召回率提升了2.4个百分点,验证了改进算法对小目标检测任务的有效性。

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