Automobile Fine-Grained Detection Algorithm Based on Multi-Improved YOLOv3 in Smart Streetlights

被引:8
|
作者
Yang, Fan [1 ,2 ]
Yang, Deming [1 ]
He, Zhiming [1 ]
Fu, Yuanhua [1 ]
Jiang, Kui [1 ]
机构
[1] Univ Elect Sci & Technol China, Sch Informat & Commun Engn, Chengdu 611731, Peoples R China
[2] Inner Mongolia Normal Univ, Coll Phys & Elect Informat, Hohhot 010022, Peoples R China
关键词
smart streetlight; YOLOv3; multi-scale training; anchor clustering; label smoothing; mixup; IOU; GIOU; fine-grained classification of automobile;
D O I
10.3390/a13050114
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Upgrading ordinary streetlights to smart streetlights to help monitor traffic flow is a low-cost and pragmatic option for cities. Fine-grained classification of vehicles in the sight of smart streetlights is essential for intelligent transportation and smart cities. In order to improve the classification accuracy of distant cars, we propose a reformed YOLOv3 (You Only Look Once, version 3) algorithm to realize the detection of various types of automobiles, such as SUVs, sedans, taxis, commercial vehicles, small commercial vehicles, vans, buses, trucks and pickup trucks. Based on the dataset UA-DETRAC-LITE, manually labeled data is added to improve the data balance. First, data optimization for the vehicle target is performed to improve the generalization ability and position regression loss function of the model. The experimental results show that, within the range of 67 m, and through scale optimization (i.e., by introducing multi-scale training and anchor clustering), the classification accuracies of trucks and pickup trucks are raised by 26.98% and 16.54%, respectively, and the overall accuracy is increased by 8%. Secondly, label smoothing and mixup optimization is also performed to improve the generalization ability of the model. Compared with the original YOLO algorithm, the accuracy of the proposed algorithm is improved by 16.01%. By combining the optimization of the position regression loss function of GIOU (Generalized Intersection Over Union), the overall system accuracy can reach 92.7%, which improves the performance by 21.28% compared with the original YOLOv3 algorithm.
引用
收藏
页数:22
相关论文
共 50 条
  • [31] Ship Target Detection Algorithm Based on Improved YOLOv3 for Maritime Image
    Chen, Dehai
    Sun, Shiru
    Lei, Zhijun
    Shao, Heng
    Wang, Yuzhao
    JOURNAL OF ADVANCED TRANSPORTATION, 2021, 2021
  • [32] Real-time object detection algorithm based on improved YOLOv3
    Zhang, Xiuling
    Dong, Xiaopeng
    Wei, Qijun
    Zhou, Kaixuan
    JOURNAL OF ELECTRONIC IMAGING, 2019, 28 (05)
  • [33] CRTSII Track Slab Crack Detection Based on Improved YOLOv3 Algorithm
    Zhang, YaoXing
    Li, Wenju
    Chen, Huiling
    He, Maoxian
    2020 5TH INTERNATIONAL CONFERENCE ON MECHANICAL, CONTROL AND COMPUTER ENGINEERING (ICMCCE 2020), 2020, : 1438 - 1441
  • [34] Research on Underwater Small Target Detection Algorithm Based on Improved YOLOv3
    Li, Jianfeng
    Zhu, Yiwen
    Chen, Mingxu
    Wang, Yongling
    Zhou, Zhiquan
    2022 16TH IEEE INTERNATIONAL CONFERENCE ON SIGNAL PROCESSING (ICSP2022), VOL 1, 2022, : 76 - 80
  • [35] An Efficient Face Mask Wearing Detection Algorithm Based on Improved YOLOv3
    Zhang, Bo
    Zhang, Xiaoxia
    Li, Zhuo
    ENGINEERING LETTERS, 2022, 30 (04)
  • [36] Multiclass Fruit Detection Using Improved YOLOv3 Algorithm
    Shrawne, Seema C.
    Sawant, Jay
    Chaubal, Omkar
    Suryawanshi, Karan
    Sirwani, Diven
    Sambhe, Vijay K.
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2024, 15 (09) : 978 - 988
  • [37] Multi-Directional Scene Text Detection Based on Improved YOLOv3
    Xiao, Liyun
    Zhou, Peng
    Xu, Ke
    Zhao, Xiaofang
    SENSORS, 2021, 21 (14)
  • [38] Multi-Branch Deepfake Detection Algorithm Based on Fine-Grained Features
    Qin, Wenkai
    Lu, Tianliang
    Zhang, Lu
    Peng, Shufan
    Wan, Da
    CMC-COMPUTERS MATERIALS & CONTINUA, 2023, 77 (01): : 467 - 490
  • [39] Cotton Stubble Detection Based on Improved YOLOv3
    Yang, Yukun
    Li, Jingbin
    Nie, Jing
    Yang, Shuo
    Tang, Jiaqiang
    AGRONOMY-BASEL, 2023, 13 (05):
  • [40] SAR Ship Detection Based on Improved YOLOv3
    Chen, Dong
    Ju, Yanwei
    IET Conference Proceedings, 2020, 2020 (09): : 929 - 934