Skip to content

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.

Notifications You must be signed in to change notification settings

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

No packages published