Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
a2970b03
Commit
a2970b03
authored
Jul 05, 2017
by
derekjchow
Committed by
GitHub
Jul 05, 2017
Browse files
Merge pull request #1855 from g-ozawa/patch-1
Fix compatibility for object_detection/core/batcher.py
parents
46ef4860
0f633765
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
object_detection/core/batcher.py
object_detection/core/batcher.py
+1
-1
No files found.
object_detection/core/batcher.py
View file @
a2970b03
...
@@ -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
.
iter
items
()})
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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment