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

test=develop

上级 3046799e
...@@ -2010,9 +2010,10 @@ def box_clip(input, im_info, name=None): ...@@ -2010,9 +2010,10 @@ def box_clip(input, im_info, name=None):
output = helper.create_variable_for_type_inference(dtype=input.dtype) output = helper.create_variable_for_type_inference(dtype=input.dtype)
inputs = {"Input": input, "ImInfo": im_info} inputs = {"Input": input, "ImInfo": im_info}
helper.append_op(type="box_clip", inputs=inputs, outputs={"Output": output}) helper.append_op(type="box_clip", inputs=inputs, outputs={"Output": output})
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.
先完成此消息的编辑!
想要评论请 注册