Tensorflow Server - GlobalMediaBridge/cosmetic-server GitHub Wiki

https://yjucho1.github.io/tensorflow/tensorflow-serving/ https://www.tensorflow.org/guide/saved_model https://github.com/tensorflow/serving/blob/master/tensorflow_serving/tools/docker/Dockerfile

텐서플로우로 작성된 모델을 저장

tf.saved_model.save(모델클래스, 경로)

해당 경로에 pb파일이 생성됨

명령을 통해 tensorflow server로 배포

nohup tensorflow_model_server \
  --rest_api_port=8501 \
  --model_name=mobilenet \
  --model_base_path="/tmp/mobilenet" >server.log 2>&1

tensorflow_model_server \
 --model_base_path=/keras-and-tensorflow-serving/my_image_classifier \
 --rest_api_port=9000 --model_name=ImageClassifier

tensorflow_model_server --port=8500 --rest_api_port=8501 \
--model_name=${MODEL_NAME} --model_base_path=${MODEL_BASE_PATH}/${MODEL_NAME} \
"$@"' > /usr/bin/tf_serving_entrypoint.sh \
&& chmod +x /usr/bin/tf_serving_entrypoint.sh

배포 된 곳의 주소 http://localhost:<--rest_api_port>/v1/models/<--model_name>:predict