yeandy commented on issue #23756: URL: https://github.com/apache/beam/issues/23756#issuecomment-1289016185
To fix this, use the function below https://github.com/apache/beam/blob/8f85a6cdf8d0d71218137fbec149d13ae6fc595b/sdks/python/apache_beam/examples/inference/tfx_bsl/build_tensorflow_model.py#L71 You can pull the PR, or just copy the code directly. Make sure you have the saved_model of the TF model. Then run the following in your Python shell ``` import tensorflow as tf from apache_beam.examples.inference.tfx_bsl.build_tensorflow_model import save_tf_model path = '/my_path/ssd_mobilenet_v2_320x320_coco17_tpu-8/saved_model' ssd_mobilenet = tf.saved_model.load(path) save_tf_model_with_signature('/my_path/ssd_mobilenet_v2_320x320_coco17_tpu-8_string_signature', ssd_mobilenet, input_dtype=tf.uint8) ``` -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
