在本教程之后,我设法使用通用初始模型重新训练我的特定分类模型。我现在想按照这个步骤将它部署在谷歌云机器学习上。
我已经设法将它导出为 MetaGraph,但我无法获得正确的输入和输出。
在本地使用它,我的图表入口点是DecodeJpeg/contents:0
一个二进制格式的 jpeg 图像。输出是我的预测。
我在本地使用的代码(正在工作)是:
softmax_tensor = sess.graph.get_tensor_by_name('final_result:0')
predictions = sess.run(softmax_tensor,{'DecodeJpeg/contents:0': image_data})
输入张量应该是DecodeJpeg
? 如果我想有一个base64 image
作为输入,我需要做些什么改变?
我将输出定义为:
outputs = {'prediction':softmax_tensor.name}
非常感谢任何帮助。