Обслуживание пользовательской модели 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