Vehicle Detection and Counting using Deep Learning basedYOLO and Deep SORT Algorithm for Urban Traffic Management System

Abstract

Vehicle counting is a process to estimate traffic density on roads to assess the traffic conditions for intelligent transportation systems (ITS). Real-time traffic management systems have become popular recently due to the availability of high end cameras and technology. The present traffic management systems focus on speed detection, signal jumping, zebra crossing but not on traffic density estimation. Proposed video-based vehicle counting and tracking method using a video captured on CCTV and handheld mobile cameras. The system can be used in smart cities to create smart traffic light signals, in which duration of each signal depends on real time vehicle density in a particular lane of road. Vehicle counting is performed in two steps: the captured video is sent to You Only Look Once (YOLO) based deep learning framework to detect, count and classify the vehicles. Multi vehicular tracking is adopted using Deep SORTalgorithm to track the vehicles in video frames. Model was trained for six different classes, using Google Colaboratory. Datasets of vehicles specifically pertaining to Indian roads environment is considered for implementation. The performance of the model was analyzed, proposed model has tested and obtained an average counting accuracy of 86.56% while the average precision is 93.85%. The model can be implemented for ascertaining the traffic density on roads and this provides knowledge for infrastructural development to authorities. It can also be an integral part of smart city projects to develop intelligent and smart traffic surveillance system.