Improving Vehicle Classification and Detection with Deep Neural Networks

Loading...
Thumbnail Image

Date

2020-11

Journal Title

Journal ISSN

Volume Title

Type

Article

Publisher

Institute of Electrical and Electronics Engineers Inc.

Series Info

2020 International Symposium on Advanced Electrical and Communication Technologies, ISAECT 202025 November 2020 2020 International Symposium on Advanced Electrical and Communication Technologies;ISAECT 2020, Marrakech, 25 November 2020 - 27 November 2020, 171534

Doi

Abstract

Vehicle detection and classification are major functions of advanced driver assistance systems (ADAS). In this paper, a deep-learning approach for vehicle detection and classification is discussed and improved. More specifically, we utilized the state of the art object detection model 'YOLOV4' with a clear focus on vehicles which made the detection process more robust. We conducted real life tests on vehicle images from Egypt at El-Sahil bridge, Imbaba and El-Hussary where our improved model detects the vehicles reliably. Our approach mainly relied on using ResNet50 and VGG16 as a classification backbones for YOLOv4, we used the GTI dataset to train and fine tune both networks to get the one with the better classification accuracy. The mean average precision (MAP) increased by 6.7% for VGG (84.49% to 91.02%) and 7.35% for ResNet (85.3% to 92.653%) then we trained YOLOv4 using OpenImages and Highway data sets for vehicle detection reaching an improvement of nearly 6.3% MAP (86.5% to 92.82%). © 2020 IEEE.

Description

Scopus

Keywords

Convolutional Neural Networks, Deep Learning, Vehicle Classification, Vehicle Detection

Citation