TensorFlow save_model, возвращающий один и тот же результат для разных входных данных

Question

Я пытаюсь экспортировать сохраненную модель, используя код ниже. Он экспортирует сохраненный_модель.pb, но для любого ввода, который я даю, я получаю тот же результат. Я думаю, что что-то не так с тем, как я делаю вывод. Я очень новичок в TensorFlow, поэтому я прошу прощения, если это простая ошибка.

with tf.gfile.FastGFile("retrained_graph.pb", 'rb') as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())
    _ = tf.import_graph_def(graph_def, name='')

def export_model(saved_model_dir, final_tensor_name):
  with tf.Session() as sess:
    init = tf.global_variables_initializer()
    sess.run(init)
    with sess.graph.as_default() as graph:
      tf.saved_model.simple_save(
        sess,
        saved_model_dir,
        inputs={'image': tf.placeholder(tf.string)},
        outputs={'prediction': graph.get_tensor_by_name(final_tensor_name + ":0")}
   )

Я также использую retrain.py из следующего урока: https://github.com/BartyzalRadek/Multi-label-Inception-net, так что это может быть что-то оттуда.

Как я структурировал свой код, я включил его в отдельный файл, поэтому мне не нужно переучиваться каждый раз, когда я хочу попытаться сохранить.