| 注册
首页|期刊导航|液晶与显示|基于改进YOLO11的无人机航拍图像小目标检测算法

基于改进YOLO11的无人机航拍图像小目标检测算法

张志豪 厉小润 陈淑涵

液晶与显示2025,Vol.40Issue(6):915-930,16.
液晶与显示2025,Vol.40Issue(6):915-930,16.DOI:10.37188/CJLCD.2025-0010

基于改进YOLO11的无人机航拍图像小目标检测算法

Small object detection algorithm in UAV aerial images based on improved YOLO11

张志豪 1厉小润 1陈淑涵1

作者信息

  • 1. 浙江大学 电气工程学院,浙江 杭州 310027
  • 折叠

摘要

Abstract

Small object detection in UAV aerial images faces challenges such as small object sizes,complex backgrounds,and limited computational resources.Most existing object detection models deployed on UAVs suffer from low accuracy and struggle to achieve a good balance between detection accuracy and efficiency.To address these issues,this paper proposes a lightweight small object detection algorithm,ACFI-YOLO11(Attention-based Cross-layer Feature Interaction-YOLO11),based on YOLO11 framework.First,this paper designs a Tiny Head branch,which enhances the model's ability to detect tiny objects by introducing higher-resolution feature maps.Second,this paper proposes a novel attention-based cross-layer feature interaction module(ACFI).This module uses a layer feature aggregation(LFA)mechanism and a Transformer encoder to enable direct information exchange between the current layer and its adjacent layers.This approach addresses the limitations of the original model's neck network,where features were passed sequentially and primarily focused on the preceding layer,failing to fully explore and leverage cross-layer feature correlations.The proposed module significantly enhances the model's representational capacity.Finally,this paper introduces space-to-depth(SPD)convolution to replace traditional convolution.This reduces the model's parameters and computational cost while preserving critical spatial information during downsampling,thereby improving detection accuracy for small objects.Experimental results on the VisDrone2021 dataset show that compared to YOLO11s,ACFI-YOLO11 achieves improvements of 4.2%,3.5%,5.2%,and 4.0%in APS,APXS,mAP50,and mAP50-95,respectively,and outperforms other comparison algorithms with a mAP50-95 of 31.7%.Furthermore,comparative experiments on the UAVDT dataset validate the superiority of ACFI-YOLO11,achieving a mAP50-95 of 83.3%,significantly outperforming other state-of-the-art algorithms.These results demonstrate that ACFI-YOLO11 not only achieves a lightweight model design but also significantly enhances detection performance,providing an efficient and practical solution for small object detection in drone aerial imagery.

关键词

小目标检测/无人机/YOLO11/特征融合/Transformer

Key words

small object detection/unmanned aerial vehicle/YOLO11/feature fusion/Transformer

分类

信息技术与安全科学

引用本文复制引用

张志豪,厉小润,陈淑涵..基于改进YOLO11的无人机航拍图像小目标检测算法[J].液晶与显示,2025,40(6):915-930,16.

基金项目

浙江省"尖兵"研发攻关计划(No.2023C01129)Supported by"Jianbing"Science and Technology Plan of Zhejiang Province(No.2023C01129) (No.2023C01129)

液晶与显示

OA北大核心

1007-2780

访问量10
|
下载量0
段落导航相关论文