0

我是机器学习的新手，基本上我创建了自己的图像数据集并对它们进行训练并在 jupyter notebook 上识别图像，之后我尝试按照本教程部署这个模型

我执行

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
4

1 回答 1

1

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

于 2018-05-02T11:46:42.033 回答