State Key Laboratory of Advanced Design and Manufacture for Vehicle Body, Hunan University, Changsha 410082, China.
School of Mechanical Engineering, Yanshan University, Qinhuangdao 066004, China.
Sensors (Basel). 2023 Apr 27;23(9):4335. doi: 10.3390/s23094335.
The integrated fast detection technology for electric bikes, riders, helmets, and license plates is of great significance for maintaining traffic safety. YOLOv5 is one of the most advanced single-stage object detection algorithms. However, it is difficult to deploy on embedded systems, such as unmanned aerial vehicles (UAV), with limited memory and computing resources because of high computational load and high memory requirements. In this paper, a lightweight YOLOv5 model (SG-YOLOv5) is proposed for the fast detection of the helmet and license plate of electric bikes, by introducing two mechanisms to improve the original YOLOv5. Firstly, the YOLOv5s backbone network and the Neck part are lightened by combining the two lightweight networks, ShuffleNetv2 and GhostNet, included. Secondly, by adopting an Add-based feature fusion method, the number of parameters and the floating-point operations (FLOPs) are effectively reduced. On this basis, a scene-based non-truth suppression method is proposed to eliminate the interference of pedestrian heads and license plates on parked vehicles, and then the license plates of the riders without helmets can be located through the inclusion relation of the target boxes and can be extracted. To verify the performance of the SG-YOLOv5, the experiments are conducted on a homemade RHNP dataset, which contains four categories: rider, helmet, no-helmet, and license plate. The results show that, the SG-YOLOv5 has the same mean average precision (mAP0.5) as the original; the number of model parameters, the FLOPs, and the model file size are reduced by 90.8%, 80.5%, and 88.8%, respectively. Additionally, the number of frames per second (FPS) is 2.7 times higher than that of the original. Therefore, the proposed SG-YOLOv5 can effectively achieve the purpose of lightweight and improve the detection speed while maintaining great detection accuracy.
电动自行车、骑手、头盔和车牌的集成快速检测技术对于维护交通安全具有重要意义。YOLOv5 是最先进的单阶段目标检测算法之一。然而,由于计算负载高和内存要求高,它很难部署在内存和计算资源有限的嵌入式系统上,例如无人机 (UAV)。在本文中,通过引入两种改进原始 YOLOv5 的机制,提出了一种用于快速检测电动自行车头盔和车牌的轻量级 YOLOv5 模型 (SG-YOLOv5)。首先,通过结合包含的两个轻量级网络 ShuffleNetv2 和 GhostNet,减轻了 YOLOv5s 骨干网络和 Neck 部分的重量。其次,通过采用基于 Add 的特征融合方法,有效地减少了参数数量和浮点运算 (FLOPs)。在此基础上,提出了一种基于场景的非真实抑制方法,以消除行人和车牌对停驶车辆的干扰,然后通过目标框的包含关系定位没有头盔的骑手的车牌并进行提取。为了验证 SG-YOLOv5 的性能,在自制的 RHNP 数据集上进行了实验,该数据集包含四个类别:骑手、头盔、无头盔和车牌。结果表明,SG-YOLOv5 的平均精度 (mAP0.5)与原始模型相同;模型参数数量、FLOPs 和模型文件大小分别减少了 90.8%、80.5%和 88.8%。此外,帧率 (FPS) 比原始模型提高了 2.7 倍。因此,所提出的 SG-YOLOv5 可以有效地实现轻量化的目的,同时在保持较高检测精度的情况下提高检测速度。
Sensors (Basel). 2023-4-27
Sensors (Basel). 2023-6-22
Plants (Basel). 2023-8-23
Sensors (Basel). 2023-6-28
Math Biosci Eng. 2022-9-5
Animals (Basel). 2023-5-31
Sensors (Basel). 2023-6-26
Sensors (Basel). 2022-9-25
IEEE Trans Pattern Anal Mach Intell. 2016-6-6
IEEE Trans Pattern Anal Mach Intell. 2015-9
Cochrane Database Syst Rev. 2008-1-23