Retraining SSD-MobileNet and Faster RCNN models

The pre-trained TensorFlow Object Detection models certainly work well for some problems. But sometimes, you may need to use your own annotated dataset (with bounding boxes around objects or parts of objects that are of particular interest to you) and retrain an existing model so it can more accurately detect a different set of object classes.

We’ll use the same Oxford-IIIT Pets dataset, as documented in the TensorFlow Object Detection API site, to retrain two existing models on your local machine, instead of using Google Cloud covered in the documentation. We’ll also add an explanation for each step when needed. The following is the step-by-step guide on how to retrain a TensorFlow object detection ...

Get Intelligent Mobile Projects with TensorFlow now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.