test=develop

revert-15296-async_double_buffered_py_reader
jerrywgz 6 years ago
parent 3046799ecd
commit 334f697da9

@ -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):
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()

Loading…
Cancel
Save