Implementation of the YOLO network architecture in pytorch. Orinal paper : https://arxiv.org/abs/1506.02640
The scripts folder contains two files:
- python notebook file which can be run in colaboratory or jupyter notebook.
- 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: