服务于TensorFlow自定义模型

Question

我是机器学习的新手，基本上我创建了自己的图像数据集，并在jupyter笔记本上进行训练和识别图像，之后我尝试通过遵循this教程来部署此模型

我执行

bazel build -c opt //tensorflow_serving/example:mnist_saved_model

bazel-bin/tensorflow_serving/example/mnist_saved_model /tmp/mnist_model

它运行成功。

如何导出我自己的模型并进行部署？我的模型名是"GoogleTensorflow“

我使用以下命令创建了这个模型

python3 export_inference_graph.py 
--input_type image_tensor 
--pipeline_config_path training/ssd_mobilenet_v1_pets.config 
--trained_checkpoint_prefix training/model.ckpt-26456 
--output_directory GoogleTensorflow

Answer 1

将您的自定义培训文件夹移动到tmp文件夹，该模型中应包含版本ex1文件夹