vehicle-counting-and-speed-estimation-yolo-sort-python

This project uses YOLOv3 for Vehicle detection and SORT(Simple Online and Realtime Tracker) for vehicle tracking

This project imlements the following tasks in the project:

1. Vehicle counting
2. Lane segmentation
3. Lane change detection
4. Speed estimation
5. Dumps all these details into a CSV file

link to the original video: https://youtu.be/PSf09R3D7Lo | updated gdrive link

link to the ouput video: gdrive link

Note that there are 4 locations in the video and so the code(4 IFs), you can delete 3 and edit the first one according to your need.

To run the project:

1. Download the code or simply run:

$ git clone https://github.com/bamwani/car-counting-and-speed-estimation-yolo-sort-python 

2. To install required dependencies, run:

$ cd car-counting-and-speed-estimation-yolo-sort-python/
$ pip3 install -r requirements.txt

3. Download yolo weights file by:

$ bash download_weights

4. Make sure you change the line of detection and lane segmentation according to your video and fine tune the threshold and confidence for YOLO model

5. Run

$ Python3 main.py -input /path/to/video/file.avi -output /path/for/output/video/file.avi -yolo /path/to/YOLO/directory/

Speed Detection:

This is an interesting project, mainly due to camera shaking(maybe due to wind or whaterver the reason may be!). This camera shake results into frame flickering. Which means we can not use traditional pixel distance travelled to Km/h mapping because as each frame flickers, the centroid of the bounding box also flickers arbitrarily. Hence I tried this new method: SPEED BETWEEN TWO LINES

Speed Between Two Lines (SBTL)

This method makes some assumptions which are as follows:

1. We know the speed limit of the road for which the video is recorded/streamed.