关键词:
目标检测
无人机航拍图像
YOLOv8
特征提取
动态检测头
摘要:
针对无人机航拍图像中目标对象尺寸小、特征信息模糊,易造成漏检、误检的问题,提出一种新型改进YOLOv8的无人机航拍图像目标检测方法,记作BDI-YOLO模型。首先,引入双向特征金字塔(Bidirectional Feature Pyramid Network,BiFPN)改进颈部结构,采用双向信息传递机制和自适应特征选择机制,增强模型在航拍图像中对不同尺度特征的提取能力。其次,将检测头部替换为动态检测头(Dynamic Head,Dyhead),增强模型对远景小目标的感受野,降低漏检、误检率。最后,在原有的CIoU损失函数中引入Inner-IoU,优化为Inner-CIoU损失函数,提高模型评估预测框的质量,加强精确定位能力。在VisDrone2019数据集上的实验结果表明,与YOLOv8模型相比,BDI-YOLO模型在精度mAP@50和mAP@50:95上分别提升3.8%和2.7%,召回率提升4%,运算量降低9.4%,参数量降低28.8%,能较好的适应复杂场景下的无人机航拍图像目标检测任务。