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 条
  • [21] Optimization algorithm for fine-grained detection of loader materials based on YOLOv5
    Gu T.
    Xu W.
    Li B.
    Li Z.
    Hui X.
    He X.
    Jisuanji Jicheng Zhizao Xitong/Computer Integrated Manufacturing Systems, CIMS, 2024, 30 (01): : 239 - 252
  • [22] Fisheye Image Object Detection Based on an Improved YOLOv3 Algorithm
    Lei, Xiaodong
    Sun, Beibei
    Peng, Jinzhu
    Zhang, Fangfang
    2020 CHINESE AUTOMATION CONGRESS (CAC 2020), 2020, : 5801 - 5805
  • [23] An Improved YOLOv3 Algorithm for Small Target Detection
    Lv, Hui
    Chen, Zhe
    Huang, Shi-Hao
    Journal of Network Intelligence, 2023, 8 (02): : 491 - 504
  • [24] Improved YOLOv3 Algorithm for Ship Target Detection
    Chen, Liankai
    Li, Bangyu
    Qi, Liang
    PROCEEDINGS OF THE 39TH CHINESE CONTROL CONFERENCE, 2020, : 7288 - 7293
  • [25] Detection of Wood Surface Defects Based on Improved YOLOv3 Algorithm
    Wang, Baogang
    Yang, Chunmei
    Ding, Yucheng
    Qin, Guangyi
    BIORESOURCES, 2021, 16 (04): : 6765 - 6779
  • [26] Multitarget Tracking Algorithm Based on an Improved YOLOv3 Algorithm
    Zhang Xiangsheng
    Shen Qing
    LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (16)
  • [27] Road Lane Line Detection Method Based on Improved YOLOv3 Algorithm
    Cui W.-L.
    Wang Y.-J.
    Kang S.-Q.
    Xie J.-B.
    Wang Q.-Y.
    Mikulovich V.I.
    Zidonghua Xuebao/Acta Automatica Sinica, 2022, 48 (06): : 1560 - 1568
  • [28] An Improved Algorithm for Detecting Pneumonia Based on YOLOv3
    Yao, Shangjie
    Chen, Yaowu
    Tian, Xiang
    Jiang, Rongxin
    Ma, Shuhao
    APPLIED SCIENCES-BASEL, 2020, 10 (05):
  • [29] Strip Steel Surface Defect Detection Based on Improved YOLOv3 Algorithm
    Li W.-G.
    Ye X.
    Zhao Y.-T.
    Wang W.-B.
    Tien Tzu Hsueh Pao/Acta Electronica Sinica, 2020, 48 (07): : 1284 - 1292
  • [30] Occluded Pedestrian Detection Algorithm Based on Improved Network Structure of YOLOv3
    Liu L.
    Zheng Y.
    Fu D.
    Moshi Shibie yu Rengong Zhineng/Pattern Recognition and Artificial Intelligence, 2020, 33 (06): : 568 - 574