Search code examples
c++tensorflowobject-detectionobject-detection-api

Is there a C++ wrapper for the Tensorflow Object Detection API?


We have trained our models and tested them successfully using the provided Python scripts. However, we now want to deploy it on our website and run a web-service for the second round of tests.

Is there a C++ wrapper so that we can use to run/execute our models the same way we do with Python scripts?


Solution

  • You'll find code to run object detection on C++ here. You'll need an exported graph (.pb format), that you can get using the TF object detection API.

    The compilation used to be tricky (except if you put your project in the tensorflow directory and compile everything with bazel, but you might not want to do that). I think it's supposed to be easier now, but I don't know how; or you can follow these instructions to compile tensorflow on its own and use it in a cmake project. You have another example of runing a graph in c++ here.