Skip to content

mattducrest/yolo-car-detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 

Repository files navigation

yolo-vehicle-detection

This vehicle detection machine learning algorithm takes as input images or videos of vehicles and outputs the same images/videos and draws bounding boxes around them with a label and probability of that label being true.

You can find the dataset used on kaggle here: https://www.kaggle.com/datasets/alkanerturan/vehicledetection

The code is based off of alkan erturan's code which you can find here: https://www.kaggle.com/code/alkanerturan/objectdetection-yolov8

My main goal was to see how running a jupyter notebook locally worked, and it turned out to be very educational.

This is how the algo performed on a video I filmed: https://youtu.be/lBezAjISCm8

I hope you have fun tweaking and running this notebook on your computer.

About

Jupyter notebook in which I play around with a pretrained yolo algorithm. The algorithm takes images or videos as inputs. It draws boxes around and labels different vehicles.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors