时间戳

首页 > 解决方案 > 如何在 tensorflow 模型服务中保存 tensorflow hub 模型以进行服务

tensorflow2.0 - 如何在 tensorflow 模型服务中保存 tensorflow hub 模型以进行服务

问题描述

我想使用来自 tensorflow hub 的 tensorflow 对象检测模型来托管模型服务。我可以下载模型，但无法以模型服务格式保存。有人知道吗？

标签： tensorflow2.0keras-layertensorflow-servingtensorflow-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')

推荐阅读