计算机工程与应用 ›› 2023, Vol. 59 ›› Issue (16): 205-211.DOI: 10.3778/j.issn.1002-8331.2206-0293

• 图形图像处理 • 上一篇    下一篇

用于货物类别检测的改进YOLOv5m方法研究

孙圆,李为相,周海军   

  1. 南京工业大学 电气工程与控制科学学院,南京 211816
  • 出版日期:2023-08-15 发布日期:2023-08-15

Research on Cargo Category Detection by Improved YOLOv5m

SUN Yuan, LI Weixiang, ZHOU Haijun   

  1. College of Electrical Engineering and Control Science, Nanjing Tech University, Nanjing 211816, China
  • Online:2023-08-15 Published:2023-08-15

摘要: 针对货车入库时对货物类别检测问题,提出一种可以用于在线检测携带货物的方法。该方法基于YOLOv5m和DeepSort算法,在YOLOv5m的主干网络中以深度可分离卷积替换标准卷积,降低模型参数量,提高模型推理速度;激活函数SiLU替换为GELU,引入随机正则的思想;融入倒置残差结构,进一步提高网络性能;损失函数CIoU替换为EIoU,提高回归精度;采用AdamW优化器改善参数更新。最后通过自制数据集进行训练和实验,实验结果表明,改进后的YOLOv5m模型具有精度高、计算量小和检测速度快的特点,能够更好地满足仓储环境下的货车入库货物检测。

关键词: YOLOv5m, DeepSort, 目标检测, GELU, AdamW, EIoU

Abstract: Aiming at the problem of cargo category detection when trucks are put into storage, this paper proposes a method that can be used to detect the carried cargo online. This method is based on YOLOv5m and DeepSort algorithms. In the backbone network of YOLOv5m, the standard convolution is replaced by a depthwise separable convolution, which reduces the amount of model parameters and improves the model inference speed. The activation function SiLU is replaced by GELU, and the idea of random regularity is introduced. Residual structure is used to further improve network performance. Loss function CIoU is replaced by EIoU to improve regression accuracy. AdamW optimizer is used to improve parameter update. Finally, the self-made dataset is used for training and experiments. The experimental results show that the improved YOLOv5m model has the characteristics of higher accuracy, less calculation and faster detection, which can better meet the detection of goods in the warehouse environment.

Key words: YOLOv5m, DeepSort, objection detection, GELU, AdamW, EIoU