提交 334f697d 编写于 作者: J jerrywgz

test=develop

上级 3046799e
...@@ -2013,6 +2013,7 @@ def box_clip(input, im_info, name=None): ...@@ -2013,6 +2013,7 @@ def box_clip(input, im_info, name=None):
return output return output
def multiclass_nms(bboxes, def multiclass_nms(bboxes,
scores, scores,
score_threshold, score_threshold,
......
...@@ -479,6 +479,7 @@ class TestBoxClip(unittest.TestCase): ...@@ -479,6 +479,7 @@ class TestBoxClip(unittest.TestCase):
out = layers.box_clip(input_box, im_info) out = layers.box_clip(input_box, im_info)
self.assertIsNotNone(out) self.assertIsNotNone(out)
class TestMulticlassNMS(unittest.TestCase): class TestMulticlassNMS(unittest.TestCase):
def test_multiclass_nms(self): def test_multiclass_nms(self):
program = Program() program = Program()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册