README.md

YOLO version-1 implemented in pytorch

Implementation of the YOLO network architecture in pytorch. Orinal paper : https://arxiv.org/abs/1506.02640

The scripts folder contains two files:

1. python notebook file which can be run in colaboratory or jupyter notebook.
2. a python file which can be run stand alone. (the notebook version is recommended for ease of execution of different cells of the code)

Example outputs :

After performing non max suppresion on outputs and assigning classes to traffic lights, persons and vehicles:

After performing non max suppresion on outputs and assigning classes to traffic lights, persons and vehicles: