The repository contains code for object detection on the Pascal VOC dataset using Faster R-CNN.
The pretrained weights can be found here. Please replace the empty weights file in the weights folder with the downloaded file (~535 MB).
test.ipynb contains code for testing the model. A sample image is placed under the images directory. Simply run all the cells of this notebook to test the model. Here's a sample result:
The repository also contains an extension of Faster R-CNN for instance segmentation (Mask R-CNN).