"tutorials/vscode:/vscode.git/clone" did not exist on "1947d87dd77eabe5893e277a52ecf0f9eb2f1063"
Commit a2970b03 authored by derekjchow's avatar derekjchow Committed by GitHub
Browse files

Merge pull request #1855 from g-ozawa/patch-1

Fix compatibility for object_detection/core/batcher.py
parents 46ef4860 0f633765
......@@ -84,7 +84,7 @@ class BatchQueue(object):
# Remember runtime shapes to unpad tensors after batching.
runtime_shapes = collections.OrderedDict(
{(key + rt_shape_str): tf.shape(tensor)
for key, tensor in tensor_dict.iteritems()})
for key, tensor in tensor_dict.items()})
all_tensors = tensor_dict
all_tensors.update(runtime_shapes)
......
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