Обслуживание пользовательской модели TensorFlow

Я новичок в машинном обучении, в основном я создал собственный набор данных изображений и тренируюсь на них и распознаю изображения на ноутбуке Jupyter, после этого я попытался развернуть эту модель, следуя этому руководству

Я исполняю

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

1 ответ

Переместите свою пользовательскую учебную папку в папку tmp, и в этой модели должна быть папка версии ex 1