Vehicle Detection Method Based on ADE-YOLOV3 Algorithm

Open Access

Abstract: Aiming at the problem of repeated detection of YOLOV3 algorithm in vehicle detection, the ADE-YOLOV3 vehicle detection algorithm is proposed. The algorithm uses K-means clustering algorithm to determine the number of target candidate frames and aspect ratio according to the inherent width and height characteristics of the vehicle. Then, according to the results obtained by clustering, the anchor parameters are reset, which makes the ADE-YOLOV3 network have certain pertinence in vehicle detection. Finally, the migration learning method is used to improve the network structure, and the optimal weight model is obtained, which improves the training precision of the model. The experimental results show that compared with the original YOLOV3 method, the mAP is increased from 91.4% to 95%, the repeated detection rate is reduced from 5.6% to 2.1%,and the detection speed by 50fps. The detection accuracy is improved and the problem of repeated detection is effectively avoided.

Keywords: Vehicle Detection; Deep Learning; YOLOv3; K-means; Migration Learning

Yunxiang Liu, Guoqing Zhang,Yuanyuan Zhang

The Author field can not be Empty

School of Computer Science & Information Engineering Shanghai Institute of Technology Shanghai, China

The Institution field can't be Empty

Volume 6, Issue 1

Volume and Issue can't be empty

347-352

The Page Numbers field can't be Empty

2432-5465

01-06-2020

Publication Date field can't be Empty