Some utilities for tensorflow serving
Project description
serving-utils
Some python utilities when using tensorflow-serving.
Installation
From PYPI: 1. Manually install tensorflow CPU or GPU version. 2. pip install serving-utils
From Github repository: 1. git clone git@github.com:Yoctol/serving-utils.git 2. make install
Usage
Client ```python from serving_utils import Client
client = Client(addr=”localhost:8500”) client.predict(…)
or async
await client.async_predict(…) ```
Saver ```python import tensorflow as tf
from serving_utils import Saver
saver = Saver( session=tf.Session(graph=your_graph), output_dir=’/path/to/serving’, signature_def_map={ ‘predict’: tf.saved_model.signature_def_utils.predict_signature_def( inputs={‘tensor_name’: tf.Tensor…}, outputs={‘tensor_name’: tf.Tensor…}, ) }, ) saver.save(…)
## Test Run the following commands:
make lint make test
## Dev
make install ```
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for serving_utils-0.4.6-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 746df903a072dd8c9f96c7965f3f1f584a0cc0e543bff89ae705a6e002bedcba |
|
MD5 | 41f3d3bde183690486da195ec560a4bb |
|
BLAKE2b-256 | d46e4ce36e8034ddb80a5af5689e13c143be2d253f02518c01278bbe56309194 |