traffic-management's Introduction

Traffic Management - The Unconventional Way

This repository is a project under Meghalaya Police Hackathon 2020.

Overview

Recently, Meghalaya Home Minister James P.K. Sangma told the Assembly that around 1.7 lakh vehicles pass through Shillong during peak hours and around 93,000 on a normal day. According to a study by the Boston Consulting Group, India’s biggest cities are losing approximately US$22 billion (Rs. 1,62,200 crore) annually due to traffic congestion. Due to narrow roads and a large number of vehicles per capita, frequent traffic congestion occurs in Shillong. It leads to wastage of time for commuters, employees, students. Blocked traffic may interfere with the passage of emergency vehicles (eg. Ambulance, Army/Police Vehicle, etc.). Higher chance of collisions due to tight spacing and constant stopping-and-going.

Objectives

Detecting vehicles and their License Plates using road-side cameras.
Analyzing parking areas and suggesting parking slots to new vehicles using mobile app.

Prerequisites

Environment

Python >= 3.6.0
Pytorch >= 1.3.0

Get code

git clone https://github.com/mrityunjay-tripathi/traffic-management.git
cd traffic_management
pip3 install --user -r requirements.txt

Download required dataset

cd data/
bash get_traffic_data.sh

Training

Download pre-trained weights.

cd weights/
bash lpr_weights.sh
bash ps_weights.sh

Modify training parameters

Review config file config/params.cfg
Adjust your GPU device. see parallels.
Adjust other parameters.

Start training

python train.py

Optional: Visualize training

tensorboard --logdir=PATH_WHERE_TENSORBOARD_LOGS_ARE_SAVED

Test

Put your test data in data/samples/test/ then run the following command.

python test.py

Start to use

python detect.py

Acknowledgement

YOLO-v3

Author

Mrityunjay Tripathi

traffic-management's People

Contributors

Stargazers

Watchers

traffic-management's Issues

OSError: [Errno 12] Cannot allocate memory

I am getting Memory Error, as I can guess there is somewhere memory leak but I can't locate it.
This is the error I am getting.

Traceback (most recent call last):
File "detection.py", line 124, in
net = d.train(train_loader)
File "detection.py", line 55, in train
for data in trainloader:
File "/home/mrityunjay/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 278, in iter
return _MultiProcessingDataLoaderIter(self)
File "/home/mrityunjay/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 682, in init
w.start()
File "/home/mrityunjay/anaconda3/lib/python3.7/multiprocessing/process.py", line 112, in start
self._popen = self._Popen(self)
File "/home/mrityunjay/anaconda3/lib/python3.7/multiprocessing/context.py", line 223, in _Popen
return _default_context.get_context().Process._Popen(process_obj)
File "/home/mrityunjay/anaconda3/lib/python3.7/multiprocessing/context.py", line 277, in _Popen
return Popen(process_obj)
File "/home/mrityunjay/anaconda3/lib/python3.7/multiprocessing/popen_fork.py", line 20, in init
self._launch(process_obj)
File "/home/mrityunjay/anaconda3/lib/python3.7/multiprocessing/popen_fork.py", line 70, in _launch
self.pid = os.fork()
OSError: [Errno 12] Cannot allocate memory

Recommend Projects

mrityunjay-tripathi / traffic-management Goto Github PK