Please wait a minute...
浙江大学学报(工学版)  2026, Vol. 60 Issue (12): 2642-2651    DOI: 10.3785/j.issn.1008-973X.2026.12.010
计算机技术     
基于改进YOLOv8的无人机仓储盘点检测算法
袁小平(),郑新文,彭聪
中国矿业大学 信息与控制工程学院,江苏 徐州 221000
UAV-based warehouse inventory object detection algorithm using improved YOLOv8
Xiaoping YUAN(),Xinwen ZHENG,Cong PENG
School of Information and Control Engineering, China University of Mining and Technology, Jiangsu 221000, China
 全文: PDF(2161 KB)   HTML
摘要:

针对原始YOLOv8n算法在无人机仓储货物盘点任务中识别精度不足的问题,提出改进的目标检测算法. 为了提升对密集堆叠货物的识别能力,引入能够适应目标重叠情况并放宽目标框宽高比约束的Inner-EIoU作为边界框损失函数. 在特征提取部分,将C2f模块中的标准卷积替换为结合多维注意力机制的全维度动态卷积,在快速空间金字塔池化模块中采用FocalModulation方法,以增强特征表达能力. 为了减少模型参数规模并提高推理速度,使用基于LAMP剪枝准则的通道剪枝方法对模型进行轻量化处理. 在自建仓储数据集上的实验结果表明,改进后算法的mAP50达到94.2%,模型大小仅为2.8 MB,推理速度提升至135帧/s,验证了该方法在边缘计算设备上部署的可行性.

关键词: 无人机;  目标检测;  仓储盘点;  注意力机制;  轻量化模型    
Abstract:

An improved object detection algorithm was proposed aiming at the issue of insufficient recognition accuracy of the original YOLOv8n algorithm in the task of drone-based warehouse inventory. Inner-EIoU, which was adaptable to target overlap and relaxed aspect ratio constraints, was introduced as the bounding box loss function in order to enhance the recognition capability for densely stacked goods. The standard convolution in the C2f module was replaced with omni-dimensional dynamic convolution that incorporated a multi-dimensional attention mechanism in the feature extraction stage. The FocalModulation method was applied to the spatial pyramid pooling-fast module in order to improve feature representation. A channel pruning method based on the LAMP criterion was used for model lightweighting in order to reduce the parameter scale of the model and increase inference speed. The experimental results on a self-built warehouse dataset showed that the improved algorithm achieved mAP50 of 94.2%, with model size of only 2.8 MB and inference speed of 135 frames/s, demonstrating the feasibility for deployment on edge computing device.

Key words: drone    object detection    warehouse inventory    attention mechanism    lightweight model
收稿日期: 2025-09-30 出版日期: 2026-10-21
CLC:  TP 391  
基金资助: 国家级大学生创新训练项目(202410290039Z).
作者简介: 袁小平(1966—),男,教授,从事图像处理、深度学习方向的研究. orcid.org/0000-0002-7936-0070. E-mail:1941@cumt.edu.cn
服务  
把本文推荐给朋友
加入引用管理器
E-mail Alert
作者相关文章  
袁小平
郑新文
彭聪

引用本文:

袁小平,郑新文,彭聪. 基于改进YOLOv8的无人机仓储盘点检测算法[J]. 浙江大学学报(工学版), 2026, 60(12): 2642-2651.

Xiaoping YUAN,Xinwen ZHENG,Cong PENG. UAV-based warehouse inventory object detection algorithm using improved YOLOv8. Journal of ZheJiang University (Engineering Science), 2026, 60(12): 2642-2651.

链接本文:

https://www.zjujournals.com/eng/CN/10.3785/j.issn.1008-973X.2026.12.010    或    https://www.zjujournals.com/eng/CN/Y2026/V60/I12/2642

图 1  YOLOv8的网络结构
图 2  传统卷积与全维度动态卷积的结构对比
图 3  4种维度注意力机制的生效方式
图 4  FocalModulation网络的结构
图 5  Inner-IoU目标检测中目标框与锚框及内部区域的示意图
图 6  LAMP剪枝的流程图
数据集类别图像数量标注框数量
训练集1 7285 488
验证集4511 479
测试集239840
总计2 4187 807
表 1  数据集类别与标注框信息
模型mAP50/%P/%R/%
YOLOv8n90.2286.8988.55
YOLOv8n_DIoU89.8885.6790.26
YOLOv8n_GIoU91.4288.8188.92
YOLOv8n_EIoU90.1586.8789.67
YOLOv8n_Inner-EIoU91.4388.2688.58
表 2  损失函数的实验结果
模型mAP50/%P/%R/%Sp/106
YOLOv8n90.2286.8988.553.01
YOLOv8n_C2f-FasterBlock90.5386.8487.482.95
YOLOv8n_C2f-MSBlock90.0386.5388.262.95
YOLOv8n_C2f-ODConv91.6487.9388.813.75
表 3  注意力模块的实验结果
模型mAP50/%Sp/106F/(帧·s?1)
SPPF90.223.01118.4
FocalModulation91.603.11117.5
SPPELAN91.053.18110.9
表 4  SPPF模块的实验结果
实验编号Inner-EioUC2f-ODConvFocalModulationmAP50/%Sp/106F/(帧·s?1)Sm/MB
0———90.223.01118.46.0
1√——91.433.01116.46.0
2√√—92.313.751007.1
3√√√93.713.86102.57.3
表 5  YOLOv8n消融实验结果
图 7  YOLOv8n原模型与改进后模型的mAP50训练曲线
FcSp/106FLOPs/109mAP50/%Sm/MB
1.03.017.893.77.7
1.51.275.494.22.8
1.71.024.693.22.4
2.00.994.192.72.1
2.50.813.291.71.8
3.00.742.387.41.6
表 6  LAMP剪枝在不同加速比下的性能对比实验
图 8  不同算法的mAP50比较
模型Sp/106FLOPs/109mAP50/%Sm/MB
Faster RCNN41.220891.654.9
YOLOv8n3.018.790.26.2
YOLOv8s11.2828.791.67.1
YOLOv5s7.216.088.014.1
YOLOv7-Tiny6.2105.386.623.9
YOLOv9t1.977.689.94.6
YOLOv10n2.36.791.34.8
YOLO11n2.626.691.95.0
本文方法1.275.494.22.8
表 7  不同算法的性能对比
图 9  改进前、后的模型在不同场景下的检测效果对比
1 刘飞, 孙延明, 陈娟, 等 新时代工业4.0背景下 加快制造业向智能制造转型的策略研究: 以粤港澳大湾区为例[J]. 广东经济, 2019, (3): 72- 81
Liu Fei, Sun Yanming, Chen Juan, et al Strategic research on accelerating the transformation of manufacturing industry to intelligent manufacturing under the background of new era industry 4.0: taking the Guangdong-Hong Kong-Macao greater bay as an example[J]. Guangdong Economy, 2019, (3): 72- 81
doi: 10.3969/j.issn.1005-0759.2019.03.014
2 何凡. 智能仓储管理系统的设计与实现 [D]. 西安: 西安电子科技大学, 2019: 23–24.
He Fan. Design and implementation of an intelligent warehouse management system [D]. Xi’an: Xi'an University of Electronic Science and Technology, 2019: 23–24.
3 孙其博, 刘杰, 黎羴, 等 物联网: 概念、架构与关键技术研究综述[J]. 北京邮电大学学报, 2010, 33 (3): 1- 9
Sun Qibo, Liu Jie, Li Shan, et al Internet of things: summarize on concepts, architecture and key technology problem[J]. Journal of Beijing University of Posts and Telecommunications, 2010, 33 (3): 1- 9
doi: 10.3969/j.issn.1007-5321.2010.03.001
4 王文娟 基于无线射频识别技术的数字化物流仓储管理系统[J]. 科学技术与工程, 2019, 19 (2): 170- 174
Wang Wenjuan Digital logistics warehouse management system based on radio frequency identification technology[J]. Science Technology and Engineering, 2019, 19 (2): 170- 174
doi: 10.3969/j.issn.1671-1815.2019.02.029
5 袁瑞阳, 刘新宇, 刘贤栋, 等. 无人云库存盘点系统[C]//物联网与无线通信-2018年全国物联网技术与应用大会论文集. 杭州: 人民邮电出版社, 2018.
Yuan Ruiyang, Liu Xinyu, Liu Xiandong, et al. Unmanned cloud library storage point system [C]// Internet of Things and Wireless Communication-Proceedings of the 2018 National Internet of Things Technology and Application Conference. Hangzhou: Posts and Telecom Press, 2018.
6 王杰. 基于机器视觉的四旋翼无人机仓库盘点系统研究与实现 [D]. 绵阳: 西南科技大学, 2021: 45−55.
Wang Jie. Research and implementation of a four rotor unmanned aerial vehicle warehouse inventory system based on machine vision [D]. Mianyang: Southwest University of Science and Technology, 2021: 45−55.
7 赵滨淋, 陈功, 李胜 基于自适应特征融合的无人机小目标检测算法[J]. 山东农业大学学报(自然科学版), 2024, 55 (6): 814- 825
Zhao Binlin, Chen Gong, Li Sheng Uav small target detection algorithm based on adaptive feature fusion[J]. Journal of Shandong Agricultural University (Natural Science Edition), 2024, 55 (6): 814- 825
8 刘智慧, 张泉灵 大数据技术研究综述[J]. 浙江大学学报(工学版), 2014, 48 (6): 957- 972
Liu Zhihui, Zhang Quanling Research overview of big data technology[J]. Journal of Zhejiang University (Engineering Science), 2014, 48 (6): 957- 972
9 张慧, 赵东辉 基于低空经济的智能物流系统运输路径优化[J]. 物流科技, 2025, 48 (21): 84- 88
Zhang Hui, Zhao Donghui Optimization of intelligent logistics system transportation path based on low altitude economy[J]. Logistics Sci-tech, 2025, 48 (21): 84- 88
10 DHL“抢跑”无人机送货 [J]. 英才, 2014(12): 16.
DHL “rush to run” drone delivery [J]. Talents Magazine, 2014(12): 16.
11 Howard A G, Zhu M, Chen B, et al. MobileNets: efficient convolutional neural networks for mobile vision applications [PP/OL]. arXiv (2017-04-17)[2025-09-15]. https://arxiv.org/abs/1704.04861.
12 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.
13 Li C, Zhou A, Yao A. Omni-dimensional dynamic convolution [C]// 2022 International Conference on Learning Representations Conference, 2022: 3-5.
14 Yang J, Li C, Dai X, et al. Focal modulation networks [PP/OL]. arXiv (2022-03-22)[2025-09-15]. https://arxiv.org/abs/2203.11926.
15 Zhang H, Xu C, Zhang S. Inner-IoU: more effective intersection over union loss with auxiliary bounding box [PP/OL]. arXiv (2023-11-06)[2025-09-15]. https://arxiv.org/abs/2311.02877.
16 Lee J, Park S, Mo S, et al. Layer-adaptive sparsity for the magnitude-based pruning [PP/OL]. arXiv (2020-10-15)[2025-09-15]. https://arxiv.org/abs/2010.07611.
17 唐克, 魏飞鸣, 李东瀛, 等 基于改进YOLOv8的轻量化无人机图像目标检测算法[J]. 计算机工程, 2026, 52 (3): 97- 106
Tang Ke, Wei Feiming, Li Dongying, et al Lightweight target detection algorithm for UAV images based on improved YOLOv8[J]. Computer Engineering, 2026, 52 (3): 97- 106
18 Tan M, Le Q V. EfficientNet: rethinking model scaling for convolutional neural networks [PP/OL]. arXiv (2019-05-28)[2025-09-15]. https://arxiv.org/abs/1905.11946.
19 Chen Y, Yuan X, Wang J, et al YOLO-MS: rethinking multi-scale representation learning for real-time object detection[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2025, 47 (6): 4240- 4252
doi: 10.1109/TPAMI.2025.3538473
[1] 杨恒,王韶涵,董青,赵科渊,杨明亮. 基于注意力增强的跨模态多层融合网络的物体位姿估计[J]. 浙江大学学报(工学版), 2026, 60(9): 1872-1880.
[2] 火久元,姜灏,常琛. 基于改进EAST的高速公路交通标识文本检测[J]. 浙江大学学报(工学版), 2026, 60(9): 1881-1889.
[3] 李珺,丁彬彬,史维娟,杨琳. 基于多尺度特征聚合的航拍图像检测算法[J]. 浙江大学学报(工学版), 2026, 60(9): 1890-1900.
[4] 邬开俊,郑云琦,魏鼎,袁海翔. 基于YOLOv8s的轻量化航拍图像小目标检测算法[J]. 浙江大学学报(工学版), 2026, 60(9): 1912-1923.
[5] 李宏伟,丁建阳,柴志雷. 用于实时语义分割的数据高效轻量模型[J]. 浙江大学学报(工学版), 2026, 60(9): 1934-1941.
[6] 宋莉,宛袁玉,宋明黎. 基于混合空洞注意力机制的图卷积强化学习[J]. 浙江大学学报(工学版), 2026, 60(9): 1980-1990.
[7] 柳长源,赵海健,吴海滨,刘佳伟. 面向自动驾驶的轻量化交警手势识别方法[J]. 浙江大学学报(工学版), 2026, 60(8): 1678-1685.
[8] 赵明华,吕雨萱,吕佳豪,陈逸飞,石程,胡静. 基于多尺度外观运动融合的视频异常检测[J]. 浙江大学学报(工学版), 2026, 60(8): 1730-1738.
[9] 刘慧霞,王荣景,高宇轩,曹猛,盛苏英,马友鹏. 基于自适应神经网络的无信号交叉口车辆无死锁调度算法[J]. 浙江大学学报(工学版), 2026, 60(8): 1832-1840.
[10] 徐恺蔚,KHIZER BIN TALIBHafiz,曹衍龙,许源平,许志杰,宋景春. 基于光流和卷积视觉Transformer的轻量级微表情识别[J]. 浙江大学学报(工学版), 2026, 60(7): 1381-1391.
[11] 董绍江,肖涛,吕振鸣,夏浩然,罗家元,孙世政,张霞,刘超. 基于改进RT-DETR的水下色偏环境中小型生物检测[J]. 浙江大学学报(工学版), 2026, 60(7): 1404-1415.
[12] 王平,徐安之,赵洪黎,魏小源,杨富龙. 融合流体自动标注与轻量化YOLOv8n的泥石流实时检测方法[J]. 浙江大学学报(工学版), 2026, 60(7): 1416-1426.
[13] 杨艳春,李佳龙. 基于多路协同与空谱先验的高光谱与多光谱图像融合[J]. 浙江大学学报(工学版), 2026, 60(7): 1427-1437.
[14] 吴佰靖,闫光辉,马龙,程文鑫,黄亚宁. 边缘感知和跨尺度特征增强的小目标水漂垃圾检测[J]. 浙江大学学报(工学版), 2026, 60(7): 1452-1463.
[15] 于鑫淼,夏楠,江佳鸿,郝子莹,把云胜. 基于多尺度特征相似性匹配的低照度目标检测[J]. 浙江大学学报(工学版), 2026, 60(7): 1464-1474.