The project aims to detect vehicles such as cars, trucks and bicycle in given images and track them in videos. The goal is achieved using YOLOv3 object detection model and PyTorch framework. The project contains a python file containing the YOLO model for detecting vehicles and a jupyter notebook which uses the model to display the output.
Note: The project is complete, but I use it for trying new techniques hence is in active development so it might not give best results. Tried to make sure the code is working and removed/commented extra lines of code as much as possible.
- Clone the repo via URL
- Set up virtual environment:
- Create virtual environment using command
virtualenv venv - Activate the virtual environment venv environment
- Install requirements using command
pip install -r requirements.txt
- Create virtual environment using command
- Download the weights and configurations of YOLOv3
- Open Jupyter Notebook in any of the following ways:
- (via Terminal/Command Prompt/Anaconda terminal) Type the command
jupyter notebook - (via Anaconda Navigator) Open Anaconda Navigator and click Jupyter Notebook
- (via Terminal/Command Prompt/Anaconda terminal) Type the command
- Run each cell