首页|期刊导航|北京航空航天大学学报|基于特征信息补充与增强的无人机视角图像目标检测算法

基于特征信息补充与增强的无人机视角图像目标检测算法OA

Object detection algorithm for UAV viewpoint images based on feature information complementation and enhancement

中文摘要英文摘要

针对无人机(UAV)视角图像中目标尺度变化大、小尺寸目标占比高且背景噪声干扰严重等问题,提出一种基于特征信息增强与补充的无人机视角图像目标检测算法.为利用高层语义信息捕捉更加丰富的多尺度信息,提出一种多元融合空间金字塔池化(MFSPPF)方法;设计多分支语义增强(MBSE)模块,可以通过多个分支提取丰富的多尺度特征并构建多尺度特征之间的联系,从而在特征融合传递信息时防止重要特征信息丢失;提出细节特征补充(DFC)模块,将低层特征信息提取细化后得到丰富细粒度特征信息,经过特征融合传递实现对高层特征中细节信息的补充.通过在VisDrone2021数据集上进行实验,结果表明:所提算法相较于基线算法YOLOv8m,平均精度(AP)、AP50、AP75、AP(s)、AP(m)、AP(l)分别提高 3.7%、5.4%、3.9%、3.1%、4.0%、7.9%.并且所提方法在YOLOv8其他模型中同样适用.与其他算法相比,所提算法在不同交并比(IOU)阈值标准下和不同尺寸目标检测下都具有优异的检测效果,同时保持较快的检测速度,能够适用于无人机视角图像检测任务.

To address the challenges of significant object scale variation,a high proportion of small-sized objects,and severe background noise interference in images captured from unmanned aerial vehicle(UAV)viewpoint images,a novel object detection algorithm based on feature information enhancement and complementation is proposed.First,to leverage high-level semantic information for capturing richer multi-scale features,a multivariate fusion spatial pyramid pooling-fast(MFSPPF)method is introduced.Second,in order to avoid losing important information during feature fusion,a multi-branch semantic enhancement(MBSE)module is created to extract rich multi-scale features over several branches and create links between these features.In addition,the detailed feature complementation(DFC)module is proposed to extract and refine the low-level feature information to obtain rich,fine-grained feature information and achieve the complementation of the detailed information in the high-level features after feature fusion.Experiments conducted on the VisDrone2021 dataset demonstrate that the proposed algorithm outperforms the baseline YOLOv8m method,with average precision(AP),AP50,AP75,AP(s),AP(m),and AP(l)improving by 3.7%,5.4%,3.9%,3.1%,4.0%,and 7.9%,respectively.Furthermore,the proposed method is applicable to other YOLOv8 models.The proposed algorithm is appropriate for UAV perspective image detection jobs because it maintains a fast detection speed while achieving better detection performance across a range of intersection over union(IOU)thresholds and object sizes.

邬开俊;蒲卓

兰州交通大学电子与信息工程学院,兰州 730000兰州交通大学电子与信息工程学院,兰州 730000

航空航天

目标检测无人机图像YOLOv8特征增强细粒度特征

object detectionUAV imagesYOLOv8feature enhancementfine-grained feature

《北京航空航天大学学报》 2026 (5)

1445-1455,11

甘肃省自然科学基金(23JRRA913)内蒙古自治区重点研发与成果转化计划项目(2023YFSH0043)兰州交通大学重点研发项目资助(ZDYF2304)Natural Science Foundation of Gansu Province(23JRRA913)Inner Mongolia Autonomous Region Key Research and Development and Achievement Transformation Program Project(2023YFSH0043)Supported by Key Research and Development Project of Lanzhou Jiaotong University(ZDYF2304)

10.13700/j.bh.1001-5965.2024.0190

评论