返回
UCN-YOLOv5: Traffic Sign Object Detection Algorithm Based on Deep Learning
DOI:10.1109/ACCESS.2023.3322371.png)
摘要
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
期刊
IF:
3.6
论文数:
9.8W
被引数:
29.4W
机构
引用论文
Traffic Sign Detection and Recognition Using Multi-Scale Fusion and Prime Sample Attention基于多尺度融合和基本样本注意力的交通标志检测与识别
IEEE ACCESS
IF3.6
U2-Net: Going deeper with nested U-structure for salient object detectionU2-Net: 基于嵌套U结构的显著性目标检测
PATTERN RECOGNITION
IF7.6
Energy-Efficient Robot Configuration and Motion Planning Using Genetic Algorithm and Particle Swarm Optimization基于遗传算法和粒子群算法的节能机器人配置和运动规划
Energies
IF0
13Carbon nuclear magnetic resonance characterization of ethylene–propylene–1-octadecene terpolymers and comparison with ethylene–propylene–1-hexene and 1-decene terpolymers
Polymer
IF0

