Skip to content

Releases: sayakpaul/E2E-Object-Detection-in-TFLite

Sample TFRecords and pbtxt file

15 Sep 03:44
Compare
Choose a tag to compare

TensorFlow Object Detection API requires the data to be packaged in TFRecords at least if you are using the API in a non-eager mode. It also needs a .pbtxt file that defines the integer mapping of the class labels.

TFLite models for pet detection

10 Sep 10:40
Compare
Choose a tag to compare

The original model is fine-tuned from an SSD MobileNetV2 model. The training notebook is available here. The attached TFLite model was quantized using dynamic-range quantization.