Как экспортировать простой сохраненный график модели для классификации изображений с помощью Tensorflow?

Question

Как экспортировать простой сохраненный график модели для классификации изображений с помощью Tensorflow?

1

У меня есть следующий код прямо сейчас, но он не работает. Я пытаюсь взять одно изображение для ввода, и классификатор выведет некоторые теги. Тренинг работает с выводом графика, но мне нужна сохраненная модель для загрузки GCP.

inputs = tf.placeholder(tf.image)
outputs = tf.placeholder(tf.string)

tf.saved_model.simple_save(sess,
        export_dir,
        inputs={"x": inputs},
        outputs={"z": outputs})

Любая помощь приветствуется! Спасибо!

Pranshu Bansal 14 апр. 2019, в 11:28

Источник

0

У меня тоже такая же проблема
Psonthalia 14 апр. 2019, в 09:24

Теги:

python

tensorflow

tensorflow-serving

1 ответ

Ещё вопросы

sdcbr · Answer 1 · 2019-04-14T12-54-00.000Z

Если вы тренировали свою модель с помощью tf.keras:

import tensorflow as tf

# The export path contains the name and the version of the model
tf.keras.backend.set_learning_phase(0) # Ignore dropout at inference
model = tf.keras.models.load_model('./model.h5')
export_path = './1'

# Fetch the Keras session and save the model
# The signature definition is defined by the input and output tensors
# And stored with the default serving key
with tf.keras.backend.get_session() as sess:
    tf.saved_model.simple_save(
        sess,
        export_path,
        inputs={'input_image': model.input},
        outputs={t.name:t for t in model.outputs})