Импортировать график из файла в Tensorflow

Question

Я пытаюсь импортировать протобуф, опубликованный Facebook, из проекта DeepFovea, расположенного здесь: https://raw.githubusercontent.com/facebookresearch/DeepFovea/master/input_graph.pb

Вот мой код:

import tensorflow.compat.v1 as tf
from tensorflow.python.platform import gfile

tf.GraphDef.FromString(tf.gfile.Open("./input_graph.pb",'rb').read())

I получаю эту ошибку:

google.protobuf.message.DecodeError: Error parsing message

Должен ли я загружать этот protobuf другим способом?

Answer 1

После долгих поисков, оказывается, вам нужно разобрать его так:

from google.protobuf import text_format

with tf.gfile.GFile(graph_filename, "rb") as f:
    graph_def = tf.GraphDef()
    graph_str = f.read()
    text_format.Merge(graph_str, graph_def)

из примера кода здесь: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/framework/test_util_test.py#L84