Блог

Прямо сейчас у меня есть следующий код, но он не работает. Я пытаюсь ввести одно изображение для ввода, и классификатор выведет некоторые теги. Обучение работает с выводом графика, но мне нужна сохраненная модель для загрузки 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})
 

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

В случае, если вы обучили свою модель с 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})