arrow
返回

UCN-YOLOv5: Traffic Sign Object Detection Algorithm Based on Deep Learning

delete2023-01-01
delete13
delete
OA
AI
P
Peilin Liu
Z
Zhaoyang Xie
T
Taijun Li *
DOI:10.1109/ACCESS.2023.3322371delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Traffic sign detection plays an important role in traffic safety and traffic management. In view of the complex and changeable environment and detection accuracy of traffic sign detection, this paper proposes UCN-YOLOv5 model based on the framework of YOLOv5.This model first replaces a new backbone network, which uses the core module RSU of U2Net to enhance the feature extraction of the network. Then, ConvNeXt-V2 is integrated, and the C3 module of its Block and YOLOv5 network is used to construct the C3_CN2 structure. The utilization of the proposed lightweight receptive field attention module LPFAConv in the Head Section represents a potential enhancement for the extraction of receptive field features. Finally, for small targets in traffic signs, Normalized Wasserstein Distance (NWD), which is insensitive to targets of different scales, is added to calculate the position loss function to replace the IoU metric to a certain extent, which further improves the detection ability of our model for traffic signs. Experiments on the TT100K dataset show that UCNYOLOv5 has excellent detection performance. Compared with the baseline model (Y0Lov5s, YOLOV5m, YOLOV5l), it improves the Map.5 index by 5.9 %, 4.9 % and 4.6 %; in the Map.5:.95 index, it is 4.4 %, 3.5 % and 2.8 % better. Moreover, the enhanced algorithm demonstrated favorable performance on the LISA and CCTSDB2021 traffic sign datasets. This research has important value for the accurate detection of traffic sign detection, and has guiding significance for in-depth research in related fields.
Keyword:
Object detection
traffic sign detection
YOLOv5
U2Net
Convnext
RFAConv

期刊

IEEE Access 封面图
IEEE Access
IF:
3.6
论文数:
9.8W
被引数:
29.4W

机构

H
Hainan University
学者数:
2.0W
论文数: 1.2W
被引数: 1.9W
引用论文

引用论文

Real-time traffic sign detection and classification towards real traffic scene
err2020-03-02
err36
PREAI
errWu, Yiqiang; Li, Zhiyong; Chen, Ying; Nai, Ke; Yuan, Jin
err分享
err收藏
U2-Net: Going deeper with nested U-structure for salient object detectionU2-Net: 基于嵌套U结构的显著性目标检测
err2020-10-01
err1.2K
errOAAI
errQin, Xuebin; Zhang, Zichen; Huang, Chenyang; Dehghan, Masood; Zaiane, Osmar R.; Jagersand, Martin
err分享
err收藏
Real-Time Detection Method for Small Traffic Signs Based on Yolov3
err2020-01-01
err71
errOAAI
errZhang, Huibing; Qin, Longfei; Li, Jun; Guo, Yunchuan; Zhou, Ya; Zhang, Jingwei; Xu, Zhi
err分享
err收藏
学者 查看更多内容