原文传递
A Novel Lightweight Traffic Sign Recognition Model Based on YOLOv5
| 题名： | A Novel Lightweight Traffic Sign Recognition Model Based on YOLOv5 |
| 正文语种： | eng |
| 作者： | Wenju Li;Gan Zhang;Liu Cui |
| 作者单位： | School of Computer Science and Information Engineering Shanghai Institute of Technology Shanghai 201418 China;School of Computer Science and Information Engineering Shanghai Institute of Technology Shanghai 201418 China;School of Computer Science and Information Engineering Shanghai Institute of Technology Shanghai 201418 China |
| 关键词： | You Only Look Once version 5 (YOLOv5); Traffic sign recognition; Coordinate attention; Lightweight; Label assignment; Loss function |
| 摘要： | How to strike a balance between detection speed and recognition accuracy has become a major challenge in real-time object detection. In this research, the YOLOv5 (You Only Look Once version 5) model was lightweight and optimized to improve the detection speed and accuracy of the network. To prune the backbone and neck are to simplify the network structure and reduce the parameters. The lightweight structure of the C3 module was designed and incorporated into the attention mechanism to improve the feature extraction capability of the network. For the public traffic sign dataset, the label assignment strategy and the loss function of YOLOv5 were refined to alleviate the imbalance between positive and negative samples and to better compute the loss, resulting in more stable and efficient training. Compared with other mainstream single-stage models, it achieves a better trade-off between speed and accuracy. With only 0.85 M parameters, 91.9% of mAP (mean average precision) and 360 FPS (frames per second) were achieved, which were 16.26% mAP and 26.67 FPS higher than the conventional YOLOv5n, respectively. The performance of our lightweight model in traffic sign detection far exceeds the most advanced achievements. |
| 出版年： | 2023 |
| 期刊名称： | Journal of Transportation Engineering |
| 卷： | 149 |
| 期： | 5 |
| 页码： | 04023025.1-04023025.11 |
相关文献
- Modeling of Vehicle Left-Turn Trajectories and Exit Lane Selection at Signalized Intersections
- Model-based dynamic toll pricing scheme for a congested suburban freeway with multiple access locations
- Analysis of In-Service Traffic Sign Visual Condition: Tree-Based Model for Mobile LiDAR and Digital Photolog Data
- Real-Time Movement-Based Traffic Volume Prediction at Signalized Intersections
- Driver Emotion Recognition Involving Multimodal Signals: Electrophysiological Response, Nasal-Tip Temperature, and Vehicle Behavior
- Self-Supervised Deep Learning Framework for Anomaly Detection in Traffic Data
检索历史
应用推荐
