Skip to content

GitLab

Commit 18d95442 authored by syiming's avatar syiming
Browse files

add different results for whether to use expected shapes or nor

parent 46c80691
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
research/object_detection/meta_architectures/faster_rcnn_meta_arch_test_lib.py
View file @ 18d95442
...@@ -582,8 +582,12 @@ class FasterRCNNMetaArchTestBase(test_case.TestCase, parameterized.TestCase): ...@@ -582,8 +582,12 @@ class FasterRCNNMetaArchTestBase(test_case.TestCase, parameterized.TestCase):
self.assertTrue(np.all(np.less_equal(anchors[:, 2], height))) self.assertTrue(np.all(np.less_equal(anchors[:, 2], height)))
self.assertTrue(np.all(np.less_equal(anchors[:, 3], width))) self.assertTrue(np.all(np.less_equal(anchors[:, 3], width)))
@parameterized.parameters(
{'use_static_shapes': False},
{'use_static_shapes': True},
)
def test_predict_shape_in_inference_mode_first_stage_only_multi_level( def test_predict_shape_in_inference_mode_first_stage_only_multi_level(
self, use_static_shapes=False): self, use_static_shapes):
batch_size = 2 batch_size = 2
height = 50 height = 50
width = 52 width = 52
......
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