计算机科学 ›› 2026, Vol. 53 ›› Issue (6A): 250800049-6.doi: 10.11896/jsjkx.250800049

• 图像处理&多媒体技术 • 上一篇    下一篇

基于改进YOLOv5的库室装备检测

周文武1,3, 雷蕾2, 铉欣3   

  1. 1 航天工程大学研究生院 北京 101400
    2 重庆电子科技职业大学通识学院 重庆 400036
    3 特种警察学院情报侦察系 北京 102211
  • 出版日期:2026-06-16 发布日期:2026-06-12
  • 通讯作者: 周文武(443882159@qq.com)

Armory Equipment Detection Based on Improved YOLOv5

ZHOU Wenwu1,3, LEI Lei2, XUAN Xin3   

  1. 1 Graduate School of Space Engineering University,Beijing 101400,China
    2 College of General Education,Chongqing Polytechnic University of Electronic Technology,Chongqing 400036,China
    3 Department of Tactical Reconnaissance and Operations,Special Police College of China,Beijing 102211,China
  • Published:2026-06-16 Online:2026-06-12
  • About author:ZHOU Wenwu,born in 1981,master.His main research interest is military equipment.

摘要: 针对部队库室装备管理效率低、实时性差的问题,以及现有检测模型在复杂库室场景下存在小目标漏检、密集目标误检、环境鲁棒性差与锚框失配等不足,提出一种基于改进YOLOv5的库室装备检测方法。以YOLOv5s为基础模型,通过构建多尺度特征增强网络,提升小目标识别能力;采用DIoU-NMS改善密集目标准确率;引入CBAM注意力机制与频域光照抑制模块,增强模型对复杂环境的适应性;通过K-means++重聚类优化锚框匹配。实验结果表明,改进模型在保持轻量化的同时,平均精度显著提升,精确率与召回率均优于原模型,可有效支撑库室装备智能化盘点与管理。

关键词: 库室装备管理, 目标检测, YOLOv5, 智能盘点

Abstract: To address the issues of low efficiency and poor real-time performance in the management of military warehouse equipment,as well as the shortcomings of existing detection models in complex warehouse scenarios-such as missed detection of small targets,false detection of dense targets,poor environmental robustness,and anchor mismatch-this paper proposes an improved YOLOv5-based detection method for warehouse equipment.Based on the YOLOv5s model,the proposed approach incorporates the following improvements:constructing a multi-scale feature enhancement network to improve the recognition capability for small targets;adopting DIoU-NMS to enhance the accuracy of dense target detection;introducing the CBAM attention mechanism and a frequency-domain illumination suppression module to strengthen the model's adaptability to complex environments;and optimizing anchor matching through K-means++ re-clustering.Experimental results show that the improved model maintains lightweight characteristics while significantly increasing average precision,with both precision and recall outperforming the original model.This method can effectively support intelligent inventory and management of warehouse equipment.

Key words: Armory equipment management, Object detection, YOLOv5, Intelligent inventory

中图分类号: 

  • TP391
[1] ZHAO Q B.Research on Intelligent Inbound and OutboundMatching Method Based on Genetic Algorithm[D].Tianjin:Tianjin University ofTechnology and Education,2024.
[2] JIN K Y,CHEN H M,CAO S R.Small Object Detection Algo-rithm Based on Improved FPN[J].China New Technologies and Products,2024(21):45-48.
[3] ZHANG Y Z,GUO W,LI W B.Omnidirectional Precise Detection Algorithm for Dense Small Objects in Remote Sensing Images[J].Journal of Jilin University(Engineering and Technology Edition),2024,54(4):1105-1113.
[4] SUN Y X.Research on Water Garbage Detection Method Using Transformer[J].Geospatial Information,2025,23(6):70-73.
[5] WU Y F.Research on Intelligent Warehouse Management System Based on Deep Learning[D].Hangzhou:Hangzhou Dianzi University,2023.
[6] ZHU X K.Research and Application of Intelligent Warehouse Management System Based on Deep Convolutional Neural Network[D].Beijing:China Academy of Electronics and Information Technology,2022.
[7] ZHANG Q.Design and Implementation of Intelligent Ware-house Management System Based on Object Detection[D].Tianjin:Tianjin University of Science and Technology,2023.
[8] HUANG X L,WEN S X.Research on Small Object Detection Model Based on YOLO Series[J].Journal of Applied Technology,2025,25(2):142-149.
[9] GUO K C.Research on Ancient Building Object Detection and Wall Material Recognition Algorithm Based on Convolutional Neural Network[D].Beijing:Beijing Forestry University,2023.
[10] CHEN M W.Research on Aerial Small Object Detection Algorithm Based on YOLOv5[D].Nanjing:Nanjing Audit University,2024.
[11] XU Z H.Research on Improved YOLOv5 Algorithm for Small Object Detection[D].Wuhan:Wuhan Polytechnic University,2025.
[12] FENG F T.Research on Small Object Detection AlgorithmBased on Improved YOLOv5s[D].Zhangjiakou:Hebei University of Architecture,2024.
[13] CHEN Y.Research on Object Detection Algorithm in Complex Scenes Based on Improved YOLOv5[D].Shijiazhuang:Hebei University of Science and Technology,2024.
[14] WANG H B.Research on Target Recognition Algorithm forAerial Images Based on YOLOv5[D].Guilin:Guilin University of Technology,2025.
[15] GENG S S.Research on Underwater Object Detection Algorithm Based on Improved YOLOv5s[D].Huainan:Anhui University of Science and Technology,2025.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!