Change graph_def file opening mode from to 'rb' (tensorflow#6071)

As discussed here: tensorflow/tensorflow#11312, this change allows this script to work in Python 3.
liorzilp · Feb 11, 2019 · c928554 · c928554
1 parent d58f759
commit c928554
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...etection/inference/detection_inference.py → ...tection/inference/deetection_inference.py b/...etection/inference/detection_inference.py → ...tection/inference/deetection_inference.py
@@ -65,7 +65,7 @@ def build_inference_graph(image_tensor, inference_graph_path):
  detected_labels_tensor: Detected labels. Int64 tensor,
  shape=[num_detections]
  """
- with tf.gfile.Open(inference_graph_path, 'r') as graph_def_file:
+ with tf.gfile.Open(inference_graph_path, 'rb') as graph_def_file:
  graph_content = graph_def_file.read()
  graph_def = tf.GraphDef()
  graph_def.MergeFromString(graph_content)