search

RidgeRun / gst-inference

A GStreamer Deep Learning Inference Framework
GNU Lesser General Public License v2.1
121 stars 29 forks source link

Question: how can we create new models ? #239

Closed boxerab closed 4 years ago

boxerab commented 4 years ago

How are the models in the model zoo generated ? Can these models be converted to run with gst-inference ?

Thanks!

michaelgruner commented 4 years ago

They are standard Tensorflow frozen binary graphs. So yes, every model in the Tensorflow zoo should be usable. We have documented the steps to freeze a graph in our R2I wiki: Generating a TF model for R2I.

michaelgruner commented 4 years ago

Just checked the TF Zoo, looks like all the networks already provide a frozen version of the model. Look for the one suffixed as .pb. For example, I downloaded ssd_mobilenet_v1_coco_2018_01_28 and inside I just found frozen_inference_graph.pb, which is what you'd use with GstInference.

boxerab commented 4 years ago

Perfect, thanks!