问题描述
我想使用来自 tensorflow hub 的 tensorflow 对象检测模型在模型服务中托管。 我可以下载模型,但无法以模型服务格式保存。 有人知道吗?
解决方法
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')