关键词:
YOLOv8
航拍图像
目标检测
注意力机制
摘要:
针对无人机航拍图像小目标检测精度低的问题,提出一种改进YOLOv8s的小目标检测算法。算法网络中,通过增加协调注意力机制(Coordinate Attention,CA)来提升目标特征提取能力;改进C2f模块并在颈部网络中引入CARAFE上采样算子,从而降低模型参数,增强多尺度特征融合能力。同时,添加一个160×160尺度的小目标检测层去替换大目标检测层,减少网络深度,提高模型对小目标的敏感度,解决误检漏检问题。实验结果表明:在VisDrone2019数据集上,相较于原YOLOv8s,改进算法的精确率P提升了6.95%,召回率R提升了8.36%,平均精度mAP50提升了13.06%,并且参数量降低了73.02%。改进算法在小目标检测方面展现出良好性能,在多项指标上优于其他YOLO系列算法,体现了其在小目标检测领域的竞争力。