object-detection-tfod

Object detection script written using Tensorflow's object detection API, using a pretrained faster-RCCN model.

Dataset trained on: Roboflow mask wearing dataset.

LabelImg was used to label and annotate images.

This was the directory in google drive used in colab.

The model used was Faster R-CNN ResNet50 V1 640x640

Edit 1: Yolov3.ipynb is a model trained using Yolov3 on the same dataset. Data cleaning and preparation scripts were used from here.