Skip to content

GitLab

Commit ef4f89e3 authored by Fan Yang's avatar Fan Yang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 468798729
parent 9c5ffc12
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
official/projects/qat/vision/serving/export_saved_model.py
View file @ ef4f89e3
...@@ -65,7 +65,7 @@ _PARAMS_OVERRIDE = flags.DEFINE_string( ...@@ -65,7 +65,7 @@ _PARAMS_OVERRIDE = flags.DEFINE_string(
'params_override', '', 'params_override', '',
'The JSON/YAML file or string which specifies the parameter to be overriden' 'The JSON/YAML file or string which specifies the parameter to be overriden'
' on top of `config_file` template.') ' on top of `config_file` template.')
_BATCH_SIZSE = flags.DEFINE_integer('batch_size', None, 'The batch size.') _BATCH_SIZE = flags.DEFINE_integer('batch_size', None, 'The batch size.')
_IMAGE_TYPE = flags.DEFINE_string( _IMAGE_TYPE = flags.DEFINE_string(
'input_type', 'image_tensor', 'input_type', 'image_tensor',
'One of `image_tensor`, `image_bytes`, `tf_example` and `tflite`.') 'One of `image_tensor`, `image_bytes`, `tf_example` and `tflite`.')
...@@ -114,14 +114,14 @@ def main(_): ...@@ -114,14 +114,14 @@ def main(_):
module = export_module_cls( module = export_module_cls(
params=params, params=params,
batch_size=_BATCH_SIZSE.value, batch_size=_BATCH_SIZE.value,
input_image_size=input_image_size, input_image_size=input_image_size,
input_type=_IMAGE_TYPE.value, input_type=_IMAGE_TYPE.value,
num_channels=3) num_channels=3)
export_saved_model_lib.export_inference_graph( export_saved_model_lib.export_inference_graph(
input_type=_IMAGE_TYPE.value, input_type=_IMAGE_TYPE.value,
batch_size=_BATCH_SIZSE.value, batch_size=_BATCH_SIZE.value,
input_image_size=input_image_size, input_image_size=input_image_size,
params=params, params=params,
checkpoint_path=_CHECKPOINT_PATH.value, checkpoint_path=_CHECKPOINT_PATH.value,
......
official/vision/serving/export_saved_model.py
View file @ ef4f89e3
...@@ -63,7 +63,7 @@ _PARAMS_OVERRIDE = flags.DEFINE_string( ...@@ -63,7 +63,7 @@ _PARAMS_OVERRIDE = flags.DEFINE_string(
'params_override', '', 'params_override', '',
'The JSON/YAML file or string which specifies the parameter to be overriden' 'The JSON/YAML file or string which specifies the parameter to be overriden'
' on top of `config_file` template.') ' on top of `config_file` template.')
_BATCH_SIZSE = flags.DEFINE_integer('batch_size', None, 'The batch size.') _BATCH_SIZE = flags.DEFINE_integer('batch_size', None, 'The batch size.')
_IMAGE_TYPE = flags.DEFINE_string( _IMAGE_TYPE = flags.DEFINE_string(
'input_type', 'image_tensor', 'input_type', 'image_tensor',
'One of `image_tensor`, `image_bytes`, `tf_example` and `tflite`.') 'One of `image_tensor`, `image_bytes`, `tf_example` and `tflite`.')
...@@ -101,7 +101,7 @@ def main(_): ...@@ -101,7 +101,7 @@ def main(_):
export_saved_model_lib.export_inference_graph( export_saved_model_lib.export_inference_graph(
input_type=_IMAGE_TYPE.value, input_type=_IMAGE_TYPE.value,
batch_size=_BATCH_SIZSE.value, batch_size=_BATCH_SIZE.value,
input_image_size=[int(x) for x in _INPUT_IMAGE_SIZE.value.split(',')], input_image_size=[int(x) for x in _INPUT_IMAGE_SIZE.value.split(',')],
params=params, params=params,
checkpoint_path=_CHECKPOINT_PATH.value, checkpoint_path=_CHECKPOINT_PATH.value,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment