基于NATCA-Greater YOLO的航拍小目标检测

作者：艾振华; 臧升睿; 陈敏; 陈倩倩; 迟洁茹; 杨国为; 于腾^*

来源：青岛大学学报(工程技术版), 2023, 38(02): 18-25.

DOI：10.13306/j.1006-9798.2023.02.003

• NAT
• CA
• Meta-ACON
• 小目标检测

为解决通用目标检测模型在无人机航拍场景下存在的物体尺度变化剧烈及复杂的背景干扰等问题，本文主要对基于NATCA-Greater YOLO的航拍小目标检测进行研究。在特征提取网络的最后一层，加入邻域注意力转换器(neighborhood attention transformer, NAT),以保留足够的全局上下文信息，并提取更多不同的特征。同时，在特征融合网络(Neck)部分，加入坐标注意力(coordinate attention, CA)模块，以获取通道信息和更长范围的位置信息，将原卷积块中的激活函数替换为Meta-ACON,并使用NAT作为新网络的预测层，以VisDrone2019-DET目标检测数据集为基准，在VisDrone2019-DET-test-dev数据集上进行测试。为了评估NATCA-Greater YOLO模型在航拍小目标检测任务中的有效性，采用Faster R-CNN、RetinaNet和单步多框目标检测(single shot multiBox detector, SSD)等检测网络在测试集上进行对比检测。研究结果表明，NATCA-Greater YOLO检测的平均精度为42%,与最先进的检测网络TPH-YOLOv5相比，NATCA-Greater YOLO的检测精度提升了2.9%,说明该模型可以准确地定位并识别目标。该研究具有一定的创新性。

• 单位
  青岛大学