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

test=develop

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