From 334f697da9e7f21a961001a4c4171ec1e6d3186d Mon Sep 17 00:00:00 2001 From: jerrywgz Date: Wed, 30 Jan 2019 03:11:13 +0000 Subject: [PATCH] test=develop --- python/paddle/fluid/layers/detection.py | 3 ++- python/paddle/fluid/tests/test_detection.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/layers/detection.py b/python/paddle/fluid/layers/detection.py index b629f54d5..63d8bd4dc 100644 --- a/python/paddle/fluid/layers/detection.py +++ b/python/paddle/fluid/layers/detection.py @@ -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, diff --git a/python/paddle/fluid/tests/test_detection.py b/python/paddle/fluid/tests/test_detection.py index 3eab9b99e..5e21dda96 100644 --- a/python/paddle/fluid/tests/test_detection.py +++ b/python/paddle/fluid/tests/test_detection.py @@ -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() -- GitLab