微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何在 Vertex AI 中为预测输入编码二进制数据

如何解决如何在 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 举报,一经查实,本站将立刻删除。