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