如何使用Tensorflow导出简单的已保存模型图进行图像分类?

时间:2019-04-14 09:15:36

标签: python tensorflow tensorflow-serving

我现在有以下代码,但无法正常工作。我试图将一张图像作为输入,分类器将输出一些标签。培训正在处理输出图形,但是我需要保存的模型才能进行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})

感谢您的帮助!谢谢!

1 个答案:

答案 0 :(得分:0)

如果您使用x训练了模型,则:

num.index(x[0])