Lightweight Deeplearning Method for Multi-vehicle Object Recognition

被引:6
作者
Li, Xun [1 ,2 ]
Yun, Xin [1 ]
Zhao, Zhengfan [3 ]
Zhang, Kaibin [1 ,2 ]
Wang, Xiaohua [1 ,2 ]
机构
[1] Xian Polytech Univ, Sch Elect & Informat, 58 Shangu Rd, Xian, Shaanxi, Peoples R China
[2] Xian Polytech Univ, Branch Shaanxi Artificial Intelligence Joint Lab, 19 Jinhua Rd, Xian, Shaanxi, Peoples R China
[3] Minist Ind & Informat Technol, Reliabil Data Ctr, Inst Elect 5, Guangzhou, Guangdong, Peoples R China
来源
INFORMATION TECHNOLOGY AND CONTROL | 2022年 / 51卷 / 02期
基金
中国国家自然科学基金;
关键词
Multi-object recognition; YOLOv4; GhostNet; Depthwise separable convolution;
D O I
10.5755/j01.itc.51.2.30667
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The recognition method based on deep learning has a large amount of calculation for the changes of different traffic densities in the actual traffic environment. In this paper, an integrated recognition method YOLOv4-L is proposed for reducing computational complexity based on the YOLOv4. The characteristics of multi-lane traffic flow with different flow densities were analyzed for statistical data sets, and k-means++ clustering algorithm was used to optimize the prior frame parameters to improve the matching degree between the prior frame. GhostNet was used to replace CSPDarknet53 of original network structure of YOLOv4 as the feature extraction network. The depthwise separable convolution module was introduced to replace the original 3x3 common convolution in feature extraction network, reduce model parameters and improve detection speed. The network model is further improved both with accuracy and robustness with the help of comprehensive method of Mosaic data enhancement, learning rate cosine annealing and label smoothing. Experimental results show that, Recognition speed is greatly improved at the expense of minimal recognition accuracy reduction: the recognition speed improvement value is 47.81%, 49.15%, 56.06% in detection speed (FPS), respectively in free flow, synchronous flow and blocked flow, the reduction value of accuracy is 2.21%, 0.67%,, 0.05% mAP, respectively.
引用
收藏
页码:294 / 312
页数:19
相关论文
共 34 条
  • [1] Abdulrahim K., 2016, Int. J. Appl. Eng. Res, V11, P713
  • [2] Arthur D, 2007, PROCEEDINGS OF THE EIGHTEENTH ANNUAL ACM-SIAM SYMPOSIUM ON DISCRETE ALGORITHMS, P1027
  • [3] Bochkovskiy A., ARXIV
  • [4] Gaussian YOLOv3: An Accurate and Fast Object Detector Using Localization Uncertainty for Autonomous Driving
    Choi, Jiwoong
    Chun, Dayoung
    Kim, Hyun
    Lee, Hyuk-Jae
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 502 - 511
  • [5] Xception: Deep Learning with Depthwise Separable Convolutions
    Chollet, Francois
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 1800 - 1807
  • [6] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [7] Rich feature hierarchies for accurate object detection and semantic segmentation
    Girshick, Ross
    Donahue, Jeff
    Darrell, Trevor
    Malik, Jitendra
    [J]. 2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, : 580 - 587
  • [8] GhostNet: More Features from Cheap Operations
    Han, Kai
    Wang, Yunhe
    Tian, Qi
    Guo, Jianyuan
    Xu, Chunjing
    Xu, Chang
    [J]. 2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, : 1577 - 1586
  • [9] Improving Faster R-CNN Framework for Fast Vehicle Detection
    Hoanh Nguyen
    [J]. MATHEMATICAL PROBLEMS IN ENGINEERING, 2019, 2019
  • [10] Real-time detection of uneaten feed pellets in underwater images for aquaculture using an improved YOLO-V4 network
    Hu, Xuelong
    Liu, Yang
    Zhao, Zhengxi
    Liu, Jintao
    Yang, Xinting
    Sun, Chuanheng
    Chen, Shuhan
    Li, Bin
    Zhou, Chao
    [J]. COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2021, 185