Fixed dictionary key for SavedModel

6117a4e5 · Nils Lattek · a6df5573 · 6117a4e5
Commit 6117a4e5 authored Jun 29, 2017 by Nils Lattek
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

object_detection/exporter.py object_detection/exporter.py +1 -1

No files found.
--- a/object_detection/exporter.py
+++ b/object_detection/exporter.py
@@ -287,7 +287,7 @@ def _write_saved_model(inference_graph_path, inputs, outputs,
      builder.add_meta_graph_and_variables(
          sess, [tf.saved_model.tag_constants.SERVING],
          signature_def_map={
-              'signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY':
+              signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY:
                  detection_signature,
          },
      )