Vehicle_detection_tracking_speed_estimation_front_view

Vehicle detection using YOLOv3,4,5 ---- Tracking using DeepSORT -----Speed Estimating using personal method

A basic application using 2D image to estimate the speed of other vehicles around, can be developed for autonomous car with 3D detection

Demo video

Use the package manager pip to install the environment with 'requirements.txt'.

pip install -r requirements.txt

Another way is installing the environment of Anaconda with 'environment.yml'

Dataset includes more than 5k images for training and testing
YOLOv3
YOLOv4
YOLOv5

The given link to the dataset includes train and test, so you need to download and extract it first
Then train the dataset with yolov5
Remember to replace track.py in Yolov5 + Deep Sort with PyTorch with the track_4.py in my repo.
Following the tutorials in Yolov5 + Deep Sort with PyTorch
Finally run:(Remember to get the right movie path responding to kmph, EX:test_day_10_007.MOV and kph_test_day_10_007.MOV

python track_4.py --source PATH_TO_THE_MOVIE --kph PATH_TO_THE_KMPH --save-vid --yolo_model PATH_TO_YOLO_CHECKPOINT --img 1280