How to properly serve an object detection model from Tensorflow Object Detection API?

Question

I am using Tensorflow Object Detection API(github.com/tensorflow/models/tree/master/object_detection) with one object detection task. Right now I am having problem on serving th

Answer 1

I was struggling with the exact problem. I was trying to host the pre-trained SSDMobileNet-COCO checkpoint from Tensorflow Object Detection API Zoo

Turns out I was using an old commit of tensorflow/models which happens to be the default sub-module of serving

I simply pulled the recent commit with

cd serving/tf_models git pull origin master git checkout master

After that, built the model server again.

bazel build //tensorflow_serving/model_servers:tensorflow_model_server

The error went away and I was able to get accurate predictions