The repository contains code for object detection on the MS COCO dataset using YOLO v3.
The pretrained weights can be found here. Please replace the empty weights file in the weights folder with the downloaded file (~237 MB).
test.ipynb contains code for testing the model. Sample images are placed under the images directory. Simply run all the cells of this notebook to test the model. Here's a sample result:
