哪种protobuf格式可以转换为VINO?

时间:2019-04-09 11:33:13

标签: tensorflow protocol-buffers openvino

当同时使用.pb和pbtxt格式读取网络时,如何将网络转换为VINO-两者中哪一个最有效?

frozen_graph = str(“ detection / 240x180_depth0.75_ssd_mobilenetv1 / frozen_inference_graph.pb”)

text_graph = str(“ detection / 240x180_depth0.75_ssd_mobilenetv1 / graph.pbtxt”)

cvNet = cv2.dnn.readNetFromTensorflow(冻结的图形,文本的图形)

我在上面使用哪个.pb和pbtxt? 即一个人如何支持另一个人?

2 个答案:

答案 0 :(得分:0)

链接https://medium.com/@prasadpal107/saving-freezing-optimizing-for-inference-restoring-of-tensorflow-models-b4146deb21b5将使您了解与模型关联的不同文件。简而言之,.pbtxt文件是人类可读的,仅包含图形的结构。它有助于检查是否缺少某些节点以进行调试。

.pb文件包含更多详细信息,并且在大多数情况下,它包含不同图层上的权重和偏差。因此,您需要使用.pb文件。链接http://answers.opencv.org/question/187904/readnetfromtensorflow-when-loading-customized-model/将为您提供更多详细信息。

答案 1 :(得分:0)

在您的情况下,仅需要Frozen_inference_graph.pb即可将拓扑转换为Vino模型。同样,您将需要模型的pipeline.json

转到“模型优化器”文件夹

python mo_tf.py \
    --input_model <PATH_TO_MODEL>/frozen_inference_graph.pb \
    --tensorflow_use_custom_operations_config extensions/front/tf/ssd_v2_support.json \
    --tensorflow_object_detection_api_pipeline_config <PATH_TO_MODEL>/pipeline.json \
    --input_shape [1,180,240,3]