object-detection-assignment

The repository is for the computer vision assignment on Object Detection

Yolo_V2.ipynb

Notebook for the UNet based model, trained from the start on limited CityScapes dataset with only 11 classes

Dataset

Fruits_Data_Set has the data needed to fine tune Yolo v2.

Requirements

• You need to have python 3 installed.

How to Run

• Install dependencies using: pip install -r requirements.txt

Local:

• Update the data folder paths i.e. train_folder and test_folder in block 88 of notebook and it will be good to go.