Commit 0f633765 authored by ozw's avatar ozw
Browse files

Fix compatibility for object_detection/core/batcher.py

parent 46ef4860
...@@ -84,7 +84,7 @@ class BatchQueue(object): ...@@ -84,7 +84,7 @@ class BatchQueue(object):
# Remember runtime shapes to unpad tensors after batching. # Remember runtime shapes to unpad tensors after batching.
runtime_shapes = collections.OrderedDict( runtime_shapes = collections.OrderedDict(
{(key + rt_shape_str): tf.shape(tensor) {(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 = tensor_dict
all_tensors.update(runtime_shapes) 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