标题：复杂场景下的无人机小目标检测算法
作者：郭越1，2 杨江涛1，2 刘 志1，2
作者单位：1 太原科技大学电子信息工程学院，山西 太原030024；2 太原科技大学先进控制与装备智能化山西省重点实验室，山西 太原 030024
关键字：无人机；小目标检测；注意力机制；上下文信息
摘要：针对无人机航拍图像检测在复杂场景下难以检测出目标以及对于小目标检测精度低的问题，提出一种适用于无人机航拍图像的目标检测算法。为提高小目标检测精度，引入小目标检测层和基于空洞卷积与注意力机制的特征增强模块，从多尺度特征图中提取更多特征，增强模型对小目标的识别能力。为应对复杂场景，引入前景感知模块，关联与前景相关的上下文信息，减少背景干扰。在VisDrone2021数据集上实验，与YOLOv5s相比，改进后模型mAP50提高5.3%，mAP50:95提高3.4%。在960×960分辨率下训练的模型mAP50达47.9%，比YOLOv4高4.9%，适用于无人机航拍图像检测。