首页|期刊导航|电子器件|增加下采样分支和替换解耦头的无人机图像小目标检测算法

增加下采样分支和替换解耦头的无人机图像小目标检测算法OA

A UAV Image Small Target Detection Algorithm with Added Subsampling Branch and Replaced Decoupled Heads

中文摘要英文摘要

在复杂环境下,高空无人机捕捉图像中存在大量小目标,进行实时目标检测时存在检测精度低和漏检问题,为此,提出一种增加下采样分支和替换解耦头的无人机图像小目标检测算法.在YOLOv7-tiny网络结构中的Neck部分增加下采样分支,充分利用低尺度特征图的图像信息;将原耦合头替换为解耦头,有利于提升目标的分类和定位的精度;引入SENet注意力模块,增强图像的上下文信息,增加对小目标的关注度,有效改善漏检问题;将CIoU替换为Focal EIoU以优化损失函数,提高计算效率,增加网络的鲁棒性.利用针对无人机拍摄的小目标检测的公开数据集VisDrone2019 开展对比实验,实验结果表明,改进后的算法漏检率降低且目标检测精度提高,召回率比YOLOv7-tiny算法提高 4.2 个百分点,mAP@0.5 比YOLOv7-tiny算法提高 2.9 个百分点.

In a complex environment,there are a large number of small targets in the images captured by high-altitude drones.When using YOLOv7-tiny for target detection,there are problems of low detection accuracy and missing detection.Therefore,a small object detection algorithm for UAV images is proposed,incorporating an added subsampling branch and a replaced decoupling head.A small target detec-tion layer is added to the Neck part of YOLOv7-tiny network structure to make full use of the image information for the low-scale feature map.The original coupling head is replaced with a decoupling head,which will help improve the accuracy of target classification and posi-tioning.SENet module is introduced to enhance the context information of the image and increase the attention to small objects,effectively improving the problem of missing detection.CIoU is replaced with Focal EIoU to optimize the loss function,making the calculation more efficient and increasing network robustness.A comparative experiment is carried out by using the public dataset of VisDrone2019 for small target detection captured by drones.The experimental results indicate that the improved algorithm has reduced the false negative rate,im-proved the precision of object detection,and achieved a recall rate of 4.2 percentage points higher than the YOLOv7-tiny algorithm.Addi-tionally,the mAP@0.5 has increased by 2.9 percentage points compared to the YOLOv7-tiny algorithm.

吕明艳;蒋联源;王智文;张灿龙

广西科技大学计算机科学与技术学院,广西 柳州 545006广西科技大学计算机科学与技术学院,广西 柳州 545006广西科技大学计算机科学与技术学院,广西 柳州 545006广西师范大学广西多源信息挖掘与安全重点实验室,广西 桂林 541004

信息技术与安全科学

小目标检测YOLOv7-tiny解耦头SENet注意力模块损失函数

small target detectionYOLOv7-tinydecoupled headSENet attention moduleloss function

《电子器件》 2026 (1)

111-119,9

国家自然科学基金项目(61962007,62266009)广西自然科学基金重点项目(2025GXNSFHA069159)广西自然科学基金项目(2018GXNSFAA294050)广西财经大数据重点实验室开放基金项目(FEDOP2022A06)

10.3969/j.issn.1005-9490.2026.01.017

评论