search

serizba / cppflow

Run TensorFlow models in C++ without installation and without Bazel
https://serizba.github.io/cppflow/
MIT License
779 stars 177 forks source link

How does cppflow use the faster-rcnn model？ #162

Closed swin223 closed 2 years ago

swin223 commented 2 years ago

Hello, dear developer, As a beginner, I have a series of questions. I am using faster-rcnn recently.The faster rcnn 'env' is

tensorflow-gpu 1.13.1

When I finish training the model, it will generate five files:

checkpoint
faster_rcnn_iter_10000.ckpt.data-00000-of-00001
faster_rcnn_iter_10000.ckpt.index
faster_rcnn_iter_10000.ckpt.meta
faster_rcnn_iter_10000.pkl

I want to try to load these files into cppflow. What is the best way? The .pb file was loaded in the cppflow example, but i didn't find it here.

serizba commented 2 years ago

Hi @swin223

Those files are the saved weights of the model. But in cppflow you not only need the saved weights, you need the complete saved_model including the operations and the model architecture.

You can have a look to the saved_model format here.

Hope it helps!

serizba commented 2 years ago

Closing due lack of activity.