如何解决如何在 Vertex AI 中为预测输入编码二进制数据
我在 StackOverflow 上进行了高低搜索以解决这个问题,但答案要么过时(不使用 TF 2.x),要么太复杂。
我已经创建了一个模型:
import tensorflow as tf
inputs = tf.keras.layers.Input((128,128,3))
conv1 = tf.keras.layers.Conv2D(16,(3,3))(inputs)
outputs = tf.keras.layers.Dense(64)(conv1)
model = tf.keras.models.Model(inputs=inputs,outputs=outputs)
我正在这样保存模型,并希望在 Vertex AI Predictions 上托管这个 SavedModel。
tf.saved_model.save(model,'placeholder_model')
我想使用编码为 base64
字符串的二进制数据作为预测输入。 Vertex AI 的在线预测官方文档说:
在您的 TensorFlow 模型代码中,您必须为您的模型命名别名 二进制输入和输出张量,以便它们以“_bytes”结尾。
我的问题是:如何导出我的 SavedModel
以便我可以使用 base64
字符串发送在线预测请求?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。