【发布时间】:2021-09-10 00:56:40
【问题描述】:
我想使用来自 tensorflow hub 的 tensorflow 对象检测模型来托管模型服务。 我可以下载模型,但无法以模型服务格式保存。 有人知道吗?
【问题讨论】:
标签: tensorflow2.0 keras-layer tensorflow-serving tensorflow-hub
我想使用来自 tensorflow hub 的 tensorflow 对象检测模型来托管模型服务。 我可以下载模型,但无法以模型服务格式保存。 有人知道吗?
【问题讨论】:
标签: tensorflow2.0 keras-layer tensorflow-serving tensorflow-hub
import tensorflow_hub as hub
import tensorflow as tf
detector = hub.load("https://tfhub.dev/tensorflow/ssd_mobilenet_v2/2")
tf.saved_model.save(detector, 'objmodel')
【讨论】: