Merge pull request #1810 from NilsLattek/master

Fixed dictionary key for SavedModel

Merge pull request #1810 from NilsLattek/master
Fixed dictionary key for SavedModel
1b7eb90f · derekjchow · GitHub · a6df5573 · 6117a4e5 · 1b7eb90f
Commit 1b7eb90f authored Jun 29, 2017 by derekjchow Committed by GitHub Jun 29, 2017
Hide 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,
          },
      )