From af448373c723ecea6a958d5ee831b0ff8860b715 Mon Sep 17 00:00:00 2001 From: jerrywgz Date: Wed, 16 Jan 2019 09:50:36 +0000 Subject: [PATCH] test=develop --- paddle/fluid/operators/detection/box_clip_op.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/operators/detection/box_clip_op.cc b/paddle/fluid/operators/detection/box_clip_op.cc index b185f12796..1e6ad7cbb3 100644 --- a/paddle/fluid/operators/detection/box_clip_op.cc +++ b/paddle/fluid/operators/detection/box_clip_op.cc @@ -36,7 +36,7 @@ class BoxClipOp : public framework::OperatorWithKernel { PADDLE_ENFORCE_EQ(im_info_dims.size(), 2, "The rank of Input(InputBox) in BoxClipOp must be 2"); PADDLE_ENFORCE_EQ(im_info_dims[1], 2, - "The last dimension of ImInfo must be 2"); + "The last dimension of ImInfo must be 3"); } ctx->ShareDim("InputBox", /*->*/ "OutputBox"); ctx->ShareLoD("InputBox", /*->*/ "OutputBox"); @@ -51,8 +51,8 @@ class BoxClipOpMaker : public framework::OpProtoAndCheckerMaker { "InputBox is a LoDTensor with shape [..., 4] holds 4 points" "in last dimension in format [xmin, ymin, xmax, ymax]"); AddInput("ImInfo", - "(Tensor) Information for image reshape is in shape (N, 2), " - "in format (height, width)"); + "(Tensor) Information for image reshape is in shape (N, 3), " + "in format (height, width, im_scale)"); AddOutput("OutputBox", "(LoDTensor) " "OutputBox is a LoDTensor with the same shape as InputBox" -- GitLab