基于轻量化目标检测网络的航拍视角小目标检测算法
DOI:
作者:
作者单位:

中国人民警察大学

作者简介:

通讯作者:

中图分类号:

TP391

基金项目:

公安部警用装备研发计划(2024ZB03)


Research on Small Object Detection Algorithm from Aerial View Based on YOLO11n
Author:
Affiliation:

1.China People'2.'3.s Police University

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对无人机航拍图像中目标尺度小、分布密集、纹理细节易丢失以及边界框回归不稳定等问题,提出一种基于YOLO11n的轻量化小目标检测算法。首先,在骨干网络浅层下采样阶段引入S2DResConv模块,通过空间重排与残差卷积增强小目标边缘和纹理信息的保留能力;其次,构建AFPN3Head渐进式自适应特征融合检测头,新增高分辨率P2小目标检测分支,保留P3和P4检测层,并移除对小目标贡献较小的P5检测分支,从总体参数受限条件下将计算预算转移至P2分支;同时,针对P2引入后正负样本分配可能出现的尺度不匹配问题,本文在检测头重构后同步更新TaskAlignedAssigner的stride张量与网格点生成逻辑,使P2、P3和P4分支分别以4、8和16的步长参与候选点生成与任务对齐匹配。最后,引入Shape-NWD回归损失,并在正样本匹配阶段采用Shape-IoU度量,以增强小目标定位稳定性。结果表明,在VisDrone2019数据集上,改进模型的Precision、Recall、mAP@50和mAP@50:95分别达到48.3%、38.3%、38.6%和23.1%,较基线YOLO11n分别提高4.2、5.7、5.8和4.0个百分点,参数量为2.45M。可见,所提方法在提升无人机航拍小目标检测精度的同时保持了较低模型复杂度,适用于对精度和部署代价均敏感的无人机端目标检测任务。

    Abstract:

    To solve the problems of small object scale, dense object distribution, loss of texture details and unstable bounding-box regression in unmanned aerial vehicle aerial images, a lightweight small-object detection algorithm based on YOLO11n is proposed. First, an S2DResConv module is introduced into the shallow downsampling stage of the backbone network. Spatial rearrangement and residual convolution are used to preserve more edge and texture information of small objects. Second, an AFPN3Head progressive adaptive feature fusion detection head is constructed. The high-resolution P2 branch is added, P3 and P4 are retained, and the P5 branch with limited contribution to small objects is removed so that the limited computational budget can be transferred to the P2 branch. After this reconstruction, the stride tensor and grid-point generation in the TaskAlignedAssigner are synchronously updated, and P2, P3 and P4 participate in candidate generation with strides of 4,8 and 16, respectively. Finally, Shape-NWD regression loss is introduced, and Shape-IoU is used in positive sample matching to enhance localization stability. The results show that, on the VisDrone2019 dataset, the Precision, Recall, mAP@50 and mAP@50:95 of the improved model reach 48.3%, 38.3%, 38.6% and 23.1%, respectively, which are 4.2, 5.7, 5.8 and 4.0 percentage points higher than those of the baseline YOLO11n. The number of parameters is 2.45M. It is demonstrated that the proposed method improves the detection accuracy of small objects in UAV aerial images while maintaining low model complexity, which is suitable for UAV-side object detection scenarios sensitive to both accuracy and deployment cost.

    参考文献
    相似文献
    引证文献
引用本文

张志刚. 基于轻量化目标检测网络的航拍视角小目标检测算法[J]. 科学技术与工程, , ():

复制
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2026-05-13
  • 最后修改日期:2026-06-06
  • 录用日期:2026-07-27
  • 在线发布日期:
  • 出版日期:
×
2026年会通知 | “技术经济学驱动智能经济生态构建与治理变革”——中国技术经济学会第三十三届学术年会(2026)会议通知暨征文启事(第一轮)
亟待确认版面费归属稿件,敬请作者关注