Tensorflow: ошибка при попытке восстановить модель в файле pb

Я пытаюсь загрузить уже обученную модель, взятую с https://github.com/tensorflow/models/tree/master/official/resnet, но когда я пытаюсь загрузить .pb, я получаю сообщение об ошибке в методе ParseFromString:

import tensorflow as tf
from tensorflow.python.platform import gfile
GRAPH_PB_PATH = '../resnet_v2_fp32_savedmodel_NHWC/1538687283/saved_model.pb'
with tf.gfile.FastGFile(GRAPH_PB_PATH, "rb") as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())
    g_in = tf.import_graph_def(graph_def, name="")
sess = tf.Session(graph=g_in)

DecodeError: Error parsing message

Что я не так?


machine-learning tensorflow resnet
person volperossa    schedule 18.03.2019    source источник

Ответы (1)


arrow_upward
1
arrow_downward

У меня была аналогичная проблема, вместо использования gfile я использую функцию tf.saved_model.loader.load, как в этом сообщении https://stackoverflow.com/a/46547595/4637693:

sess = tf.Session(graph=tf.Graph())
model = tf.saved_model.loader.load(sess, [tf.saved_model.tag_constants.SERVING], './model')
graph_def = model.graph_def
person Carlos Ramírez    schedule 28.06.2019