提交 3118a5e8 编写于 作者: J jerrywgz

refine test_detection, test=develop

上级 d9b93962
...@@ -476,7 +476,7 @@ class TestMulticlassNMS(unittest.TestCase): ...@@ -476,7 +476,7 @@ class TestMulticlassNMS(unittest.TestCase):
bboxes = layers.data( bboxes = layers.data(
name='bboxes', shape=[-1, 10, 4], dtype='float32') name='bboxes', shape=[-1, 10, 4], dtype='float32')
scores = layers.data(name='scores', shape=[-1, 10], dtype='float32') scores = layers.data(name='scores', shape=[-1, 10], dtype='float32')
output = layers.multiclass_nms(bboxes, scores, 0.3, 400, 0.7, 200) output = layers.multiclass_nms(bboxes, scores, 0.3, 400, 200, 0.7)
self.assertIsNotNone(output) self.assertIsNotNone(output)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册