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
46c80691
Commit
46c80691
authored
Aug 06, 2020
by
syiming
Browse files
add different results for whether to use expected shapes or nor
parent
2dbe7ba0
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
1 deletion
+5
-1
research/object_detection/meta_architectures/faster_rcnn_meta_arch_test_lib.py
...tion/meta_architectures/faster_rcnn_meta_arch_test_lib.py
+5
-1
No files found.
research/object_detection/meta_architectures/faster_rcnn_meta_arch_test_lib.py
View file @
46c80691
...
...
@@ -626,9 +626,13 @@ class FasterRCNNMetaArchTestBase(test_case.TestCase, parameterized.TestCase):
'rpn_box_encodings'
:
(
batch_size
,
expected_num_anchors
,
4
),
'rpn_objectness_predictions_with_background'
:
(
batch_size
,
expected_num_anchors
,
2
),
'anchors'
:
(
18300
,
4
)
}
if
use_static_shapes
:
expected_output_shapes
[
'anchors'
]
=
(
expected_num_anchors
,
4
)
else
:
expected_output_shapes
[
'anchors'
]
=
(
18300
,
4
)
if
use_static_shapes
:
results
=
self
.
execute
(
graph_fn
,
[
images
],
graph
=
g
)
else
:
...
...
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