Skip to content

GitLab

Commit 1b7eb90f authored by derekjchow's avatar derekjchow Committed by GitHub
Browse files

Merge pull request #1810 from NilsLattek/master 

Fixed dictionary key for SavedModel
parents a6df5573 6117a4e5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
object_detection/exporter.py
View file @ 1b7eb90f
...@@ -287,7 +287,7 @@ def _write_saved_model(inference_graph_path, inputs, outputs, ...@@ -287,7 +287,7 @@ def _write_saved_model(inference_graph_path, inputs, outputs,
builder.add_meta_graph_and_variables( builder.add_meta_graph_and_variables(
sess, [tf.saved_model.tag_constants.SERVING], sess, [tf.saved_model.tag_constants.SERVING],
signature_def_map={ signature_def_map={
'signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY': signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY:
detection_signature, detection_signature,
}, },
) )
......
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