Traffic sign recognition based on Attention Mechanism
DOI:
CSTR:
Author:
Affiliation:

School of Mechanical and Electrical Engineering, Central South University of Forestry and Technology, Changsha 410000, Hunan, China

Clc Number:

TP389.1

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Aiming at the problem of low accuracy of small target detection in traffic sign recognition tasks,which caused by that most of traffic signs in actual scene are small and dense,this paper proposes an improved YOLOv5 algorithm.firstly,embedding the CBAM into the Backbone and Neck of YOLOv5 network to improve the network feature extraction ability.and in order to solve the problem of slow network converge caused by GIOU Loss, DIoU Loss was used as the regression Loss function of the network. Experimental results show that the improved algorithm reaches 96.40% mAP in traffic sign recognition task,which is 6.83% higher than the original YOLOv5 algorithm. Finally, sending the improved network into TX2 embedded system to recognize traffic signs in real video,the result shows that the improved algorithm can run smoothly in embedded system.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: May 10,2024
  • Published:
Article QR Code