Please wait a minute...
浙江大学学报(工学版)  2026, Vol. 60 Issue (9): 1912-1923    DOI: 10.3785/j.issn.1008-973X.2026.09.008
计算机技术、自动控制技术     
基于YOLOv8s的轻量化航拍图像小目标检测算法
邬开俊1,2(),郑云琦1,魏鼎2,袁海翔1
1. 兰州交通大学 数理学院,甘肃 兰州 730070
2. 兰州交通大学 电子与信息工程学院,甘肃 兰州 730070
YOLOv8s based lightweight algorithm for small object detection in aerial imagery
Kaijun WU1,2(),Yunqi ZHENG1,Ding WEI2,Haixiang YUAN1
1. School of Mathematical and Physics, Lanzhou Jiaotong University, Lanzhou 730070, China
2. School of Electronic and Information Engineering, Lanzhou Jiaotong University, Lanzhou 730070, China
 全文: PDF(3699 KB)   HTML
摘要:

为了解决无人机(UAV)航拍图像目标检测面临的复杂背景干扰、目标尺寸小以及设备资源有限等挑战,提出基于YOLOv8s的轻量化小目标检测方法. 为了增强微小目标检测能力,重建高分辨率检测头. 设计动态多尺度聚合网络,动态调整感受野以适应目标尺度变化. 提出高效多尺度特征深度融合模块,自适应整合无人机图像不同层级提取的特征信息. 将融合后的特征进行双重精炼,建立高效通道-空间注意力模块. 在VisDrone2021公开数据集上的实验结果显示,改进后算法在验证集上的平均精度、不同交并比阈值下精度以及中小尺度目标检测精度均提升5.0%以上. 在保持较低计算量的同时,参数量减少16.25%. 在DOTA数据集上验证模型的泛化性,所提算法在提高检测性能的同时,实现了模型轻量化,可以应用于无人机航拍图像目标检测.

关键词: 目标检测航拍图像YOLOv8s特征融合特征提取    
Abstract:

An improved lightweight small object detection method based on YOLOv8s was proposed to address the challenges of complex background interference, small object sizes, and limited device resources in unmanned aerial vehicle (UAV) image object detection. A high-resolution detection head was reconstructed to improve the sensitivity to tiny objects. A dynamic multi-scale aggregation network was designed to adaptively adjust the receptive field for scale-variant targets, while an efficient multi-scale feature deep fusion module was proposed to integrate features from different semantic levels. An efficient channel-spatial attention module was introduced to further refine fused features. Experiments on the VisDrone2021 dataset demonstrated that the proposed method achieved improvements of over 5.0% in average precision, precision at different intersection over union thresholds, and detection precision for small and medium-sized objects, with a 16.25% reduction in parameters. Tests on the DOTA dataset validated the generalization ability of the proposed model. Results demonstrate that the proposed method improves the detection performance while achieving the model light weighting, indicating its practical value for object detection in UAV aerial images.

Key words: object detection    aerial image    YOLOv8s    feature fusion    feature extraction
收稿日期: 2025-07-20 出版日期: 2026-07-20
CLC:  TP 391  
基金资助: 甘肃省重点研发计划资助项目(25YFGA047);鄂尔多斯市重点研发计划资助项目(YF20250245).
作者简介: 邬开俊(1978—),男,教授,博士,从事计算机视觉与图像处理研究. orcid.org/0000-0002-8939-1874. E-mail:wkj@mail.lzjtu.cn
服务  
把本文推荐给朋友
加入引用管理器
E-mail Alert
作者相关文章  
邬开俊
郑云琦
魏鼎
袁海翔

引用本文:

邬开俊,郑云琦,魏鼎,袁海翔. 基于YOLOv8s的轻量化航拍图像小目标检测算法[J]. 浙江大学学报(工学版), 2026, 60(9): 1912-1923.

Kaijun WU,Yunqi ZHENG,Ding WEI,Haixiang YUAN. YOLOv8s based lightweight algorithm for small object detection in aerial imagery. Journal of ZheJiang University (Engineering Science), 2026, 60(9): 1912-1923.

链接本文:

https://www.zjujournals.com/eng/CN/10.3785/j.issn.1008-973X.2026.09.008        https://www.zjujournals.com/eng/CN/Y2026/V60/I9/1912

图 1  动态多尺度特征增强目标检测算法整体网络结构
图 2  动态多尺度聚合网络结构图
图 3  多尺度深度融合模块示意图
图 4  高效通道-空间注意力模块示意图
图 5  目标尺寸分布
方法AP/%AP50/%AP75/%APS/%APM/%APL/%FLOPs/109Params/106
BaselineP2DMSA-NetECSAMTDFM
21.937.222.212.433.441.828.511.14
23.940.124.115.634.541.630.06.60
23.239.123.413.035.044.835.114.67
23.539.624.113.935.745.647.614.35
22.538.022.912.934.242.132.311.56
24.040.424.215.434.838.931.57.14
25.542.826.016.836.542.434.77.95
25.843.126.617.037.141.535.58.19
27.245.027.718.339.043.738.39.33
表 1  VisDrone验证集上的消融实验结果
图 6  基线方法与所改进方法在VisDrone验证集上的混淆矩阵结果
方法AP/%AP50/%AP75/%APS/%APM/%APL/%FLOPs/109Params/106
YOLOv8n17.931.117.99.627.635.76.82.68
YOLOv8n+DMSA-Net+TDFM+P2+ECSAM21.335.821.812.430.937.112.22.51
YOLOv8m24.741.025.215.336.843.579.125.85
YOLOv8m+DMSA-Net+TDFM+P2+ECSAM28.446.929.019.140.145.3101.821.89
YOLOv8l26.543.327.316.239.449.4164.943.6
YOLOv8l+DMSA-Net+TDFM+P2+ECSAM30.149.430.821.042.050.1213.039.2
YOLOv8x27.444.428.417.041.147.5257.468.13
YOLOv8x+DMSA-Net+TDFM+P2+ECSAM30.750.131.521.842.647.8328.761.05
表 2  YOLOv8其他模型改进后在VisDrone验证集上的检测性能对比结果
方法AP/%AP50/%AP75/%APS/%APM/%APL/%
YOLOv5s27.742.429.410.923.641.9
YOLOv8s28.142.830.410.424.141.3
YOLOv10s27.642.729.010.724.440.3
YOLOv11s28.043.029.811.124.442.1
YOLOv12s27.742.529.510.023.641.9
本文方法30.846.332.712.428.242.8
表 3  DOTA数据集上对比实验结果
图 7  DOTA数据集检测结果可视化
模型Params/106AP/%AP50/%AP75/%APS/%APM/%APL/%FLOPs/109
RetinaNet[3]21.378.015.57.62.113.223.715.9
FasterR-CNN[4]41.7212.823.912.65.221.129.726.8
YOLOXm[24]12.4213.924.514.16.022.023.918.1
TOOD[25]15.6416.928.216.87.924.343.143.8
VFNet[26]9.8417.630.317.98.026.736.410.2
YOLOX-Tiny5.0414.827.814.97.622.127.87.6
YOLOv8m25.8519.033.219.29.029.441.779.1
YOLOv10m15.3219.534.519.69.730.041.458.9
YOLO11s[22]9.4217.631.317.88.027.236.421.3
YOLO12s[23]9.2317.631.217.88.127.435.621.2
RetinaNet-R50-FPN36.5216.427.614.86.027.442.7210
GFL[27]32.2819.132.119.29.130.040.9206
本文方法9.3320.936.921.111.031.442.838.3
表 4  VisDrone测试集上的对比实验结果
图 8  检测热力图对比
图 9  检测结果可视化对比
1 REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: unified, real-time object detection [C]// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas: IEEE, 2016: 779–788.
2 LIU W, ANGUELOV D, ERHAN D, et al. Ssd: single shot multibox detector [C]// Proceedings of the European Conference on Computer Vision. Amsterdam: Springer, 2016: 21–37.
3 LIN T Y, GOYAL P, GIRSHICK R, et al. Focal loss for dense object detection [C]// Proceedings of the IEEE/CVF International Conference on Computer Vision Workshops. Venice: IEEE, 2017: 2999–3007.
4 REN S, HE K, GIRSHICK R, et al. Faster R-CNN: towards real-time object detection with region proposal networks [C]// Conference on Neural Information Processing Systems. Montreal: MIT Press, 2015: 91–99.
5 翟亚红, 陈雅玲, 徐龙艳, 等 改进 YOLOv8s 的轻量级无人机航拍小目标检测算法[J]. 浙江大学学报: 工学版, 2025, 59 (8): 1- 10
ZHAI Yahong, CHEN Yaling, XU Longyan, et al Improved YOLOv8s lightweight small target detection algorithm of UAV aerial image[J]. Journal of Zhejiang University: Engineering Science, 2025, 59 (8): 1- 10
6 CAO Y, HE Z, WANG L, et al. VisDrone-DET2021: the vision meets drone object detection challenge results [C]// Proceedings of the IEEE/CVF International Conference on Computer Vision Workshops. Montreal: IEEE, 2021: 2847–2854.
7 YANG X, YANG J, YAN J, et al. SCRDet: towards more robust detection for small, cluttered and rotated objects [C]// Proceedings of the IEEE/CVF International Conference on Computer Vision. Seoul: IEEE, 2019: 8231–8240.
8 尹向雷, 屈少鹏, 解永芳, 等 基于渐进特征融合及多尺度空洞注意力的遮挡鸟巢检测[J]. 浙江大学学报: 工学版, 2025, 59 (3): 535- 545
YIN Xianglei, QU Shaopeng, XIE Yongfang, et al Occluded bird nest detection based on asymptotic feature fusion and multi-scale dilated attention[J]. Journal of Zhejiang University: Engineering Science, 2025, 59 (3): 535- 545
9 FENG C, CHEN Z, KOU R, et al. HazyDet: open-source benchmark for drone-view object detection with depth-cues in Hazy scenes [J]. Computing Research Repository, 2024, 2409: 19833.
10 MA Z, LUO P, SHEN X LMSOE-Net: lightweight multi-scale small object enhancement network for UAV aerial images[J]. Complex and Intelligent Systems, 2025, 11 (8): 333
doi: 10.1007/s40747-025-01971-0
11 LI Y, LI Q, PAN J, et al SOD-YOLO: small-object-detection algorithm based on improved YOLOv8 for UAV images[J]. Remote Sensing, 2024, 16 (16): 3057
doi: 10.3390/rs16163057
12 LI H, QU H. DASSF: dynamic-attention scale-sequence fusion for aerial object detection [C]// Conference on Computational Visual Media. Hong Kong: Springer, 2025: 212–227.
13 GHARATAPPEH S, SEKEH S, DHIMAN V. Weather-aware object detection transformer for domain adaptation [J]. Computing Research Repository, 2025, 2504: 10877.
14 NIKOUEI M, BAROUTIAN B, NABAVI S, et al Small object detection: a comprehensive survey on challenges, techniques and real-world applications[J]. Computing Research Repository, 2025, 2503: 20516
15 MA X, DAI X, BAI Y, et al. Rewrite the stars [C]// Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Seattle: IEEE, 2024: 5694-5703.
16 LI W, CHEN H, LIU Q, et al Attention mechanism and depthwise separable convolution aided 3DCNN for hyperspectral remote sensing image classification[J]. Remote Sensing, 2022, 14 (9): 2215
doi: 10.3390/rs14092215
17 WANG Q, WU B, ZHU P, et al. ECA-Net: efficient channel attention for deep convolutional neural networks [C]// Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Seattle: IEEE, 2020: 11531–11539.
18 YANG L, ZHANG R Y, LI L, et al. SimAM: a simple, parameter-free attention module for convolutional neural networks[C]//Proceedings of the International Conference on Machine Learning. [S.l.]: ACM, 2021: 11863–11874.
19 ZHU P, WEN L, DU D, et al Detection and tracking meet drones challenge[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2022, 44 (11): 7380- 7399
doi: 10.1109/TPAMI.2021.3119563
20 XIA G S, BAI X, DING J, et al. DOTA: a large-scaledataset for object detection in aerial images[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Salt Lake City: IEEE, 2018: 3974–3983.
21 WANG A, CHEN H, LIU L, et al. YOLOv10: real-time end-to-end object detection [C]// Conference on Neural Information Processing Systems. Vancouver: MIT Press, 2024: 107984–108011.
22 KHANAM R, HUSSAIN M. YOLOv11: an overview of the key architectural enhancements [EB/OL]. [2026-07-10]. https://arxiv.org/abs/2410.17725.
23 TIAN Y, YE Q, DOERMANN D Yolov12: attention-centric real-time object detectors[J]. Advances in Neural Information Processing Systems, 2026, 38: 78433- 78457
24 GE Z, LIU S, WANG F, et al. YOLOX: exceeding YOLO series in 2021 [EB/OL]. [2026-07-10]. https://arxiv.org/abs/2107.08430.
25 FENG C, ZHONG Y, GAO Y, et al. TOOD: task-aligned one-stage object detection [C]// Proceedings of the 2021 IEEE/CVF International Conference on Computer Vision. Montreal: IEEE, 2021: 3490–3499.
26 ZHANG H, WANG Y, DAYOUB F, et al. VarifocalNet: an iou-aware dense object detector [C]// Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. [S.l.]: IEEE, 2021: 8514–8523.
[1] 连远锋,范树玉,张本哲,王森. 融合几何约束特征提取与多奖励协同优化的仿生步态控制[J]. 浙江大学学报(工学版), 2026, 60(9): 1862-1871.
[2] 火久元,姜灏,常琛. 基于改进EAST的高速公路交通标识文本检测[J]. 浙江大学学报(工学版), 2026, 60(9): 1881-1889.
[3] 李珺,丁彬彬,史维娟,杨琳. 基于多尺度特征聚合的航拍图像检测算法[J]. 浙江大学学报(工学版), 2026, 60(9): 1890-1900.
[4] 武晓春,李梓宁. 基于YOLOv11n的改进铁路工人安全穿戴检测模型[J]. 浙江大学学报(工学版), 2026, 60(9): 1901-1911.
[5] 陈广秋,任天蓉,段锦,黄丹丹. 结合边缘辅助与多级特征融合的跨模态语义分割算法[J]. 浙江大学学报(工学版), 2026, 60(8): 1782-1791.
[6] 柳长源,赵海健,吴海滨,刘佳伟. 面向自动驾驶的轻量化交警手势识别方法[J]. 浙江大学学报(工学版), 2026, 60(8): 1678-1685.
[7] 董绍江,肖涛,吕振鸣,夏浩然,罗家元,孙世政,张霞,刘超. 基于改进RT-DETR的水下色偏环境中小型生物检测[J]. 浙江大学学报(工学版), 2026, 60(7): 1404-1415.
[8] 王平,徐安之,赵洪黎,魏小源,杨富龙. 融合流体自动标注与轻量化YOLOv8n的泥石流实时检测方法[J]. 浙江大学学报(工学版), 2026, 60(7): 1416-1426.
[9] 吴佰靖,闫光辉,马龙,程文鑫,黄亚宁. 边缘感知和跨尺度特征增强的小目标水漂垃圾检测[J]. 浙江大学学报(工学版), 2026, 60(7): 1452-1463.
[10] 于鑫淼,夏楠,江佳鸿,郝子莹,把云胜. 基于多尺度特征相似性匹配的低照度目标检测[J]. 浙江大学学报(工学版), 2026, 60(7): 1464-1474.
[11] 王铮,张梦君,姜楠,王万良,屠杭垚. 基于多特征融合和牛顿-拉夫逊优化算法的LSTM日径流预测[J]. 浙江大学学报(工学版), 2026, 60(7): 1567-1576.
[12] 张乃洲,赵云超,曹薇,张啸剑. 基于多视图跨模态特征融合的图像描述生成[J]. 浙江大学学报(工学版), 2026, 60(6): 1205-1212.
[13] 董博,吕东澔,喻大华,杜晓炜. 融合多域特征的VAE模型在肌肉疲劳分析中的应用[J]. 浙江大学学报(工学版), 2026, 60(6): 1317-1328.
[14] 李国燕,于威,梅玉鹏,张明辉,王新强. 全局局部特征融合的遥感图像建筑物提取[J]. 浙江大学学报(工学版), 2026, 60(5): 1100-1108.
[15] 于天河,王文龙,刘镛,杨壮壮,侯善冲. 改进的有雾图像中被遮挡车辆及行人识别算法[J]. 浙江大学学报(工学版), 2026, 60(4): 738-750.