Skip to content

Measure traffic density using Inception Resnet & Faster RCNN

Notifications You must be signed in to change notification settings

zhiaozhou/Vehicle_Count_Deep_Learning

Repository files navigation

Vehicle_Count_CNN

Team Members

Member Email Github
Zhiao Zhou [email protected] https://github.com/zhiaozhou
Julian Ferreiro [email protected] https://github.com/jif245
Lingyi Zhang [email protected] https://github.com/lingyielia

Description

Traffic count is the fundamental element of transportation-related studies. Although traffic counting has been widely studied, more effort is on annually traffic estimation based on collected daily traffic counts, rather than improving accuracy and efficiency of the raw data collection. Moreover, most of people are still collecting traffic density data manually instead of using machines. We successfully implemented two CNN-based models (classification and object detection) for detecting vehicle numbers in images: Inception Resnet V2 (Model 1) and Faster RCNN with Inception Resnet V2 (Model 2). In model 1, Inception Resnet V2 was used to classify the vehicle counts ranging from 0 to 41. In model 2, we first conducted detection of cars using Faster RCNN with Inception Resnet V2 and then count the frequency of detections in each image. In this way, traffic counts can be conducted automatically and in real-time using pre-installed traffic cameras, which could largely reduce expenses on labor as well as extra equipment installation and maintenance.
The main commands can be found in the main.sh script.

Demo

This project also contains a demo website that shows how our model works which is yielded by Demo_Detection.ipynb We tested our models on a traffic cam from 6th Avenue, Manhattan, New York

Drawing Drawing

Paper

You can find the pdf file of our final paper here

About

Measure traffic density using Inception Resnet & Faster RCNN

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published