|
Vehicle Identification Algorithms Based on Lightweight Neural Networks
DENG Chao1,2,3,4, MA Junjie1, YAN Yi1, WANG Youfu1, LI Yanqi1
2024, 43(4):
80-87.
DOI: 10.3969/j.issn.1674-0696.2024.04.12
lightweight neural network algorithm MobileNetV3-YOLOv5s based on hybrid attention mechanism was proposed to address the complex network structure, large parameter quantity, and high hardware requirements of current vehicle identification neural network algorithms. Firstly, the bneck module of MobileNetV3 was used to replace the backbone network of YOLOv5s. Secondly, the large convolutional kernel was replaced with a small convolutional kernel, and meanwhile, the SPPF algorithm was improved with a feature fusion method that required less computation. Finally, SENet and CAM algorithms were integrated into the backbone network to form a mixed attention module, which increased the network's weight on important regions. The experimental results show that on the UA-DETRAC dataset, the parameter amount of the proposed algorithm is reduced by 82.6% compared to YOLOv5s, only 2.34 MB, with an average identification rate of 98.2%. On the Nvidia Jetson AGX NX, the detection speed reaches 31 frames per second, with a speed increase of 10.7%. The proposed algorithm can be better deployed on edge devices and meet the requirements of autonomous driving.
References |
Related Articles |
Metrics
|