diff --git a/paddle/fluid/operators/detection/box_clip_op.cc b/paddle/fluid/operators/detection/box_clip_op.cc index 1e6ad7cbb3899b390d0fa9efcb67f10f369499cb..609bd5606b297d0d4b15c2d2dadbea061fd47ff6 100644 --- a/paddle/fluid/operators/detection/box_clip_op.cc +++ b/paddle/fluid/operators/detection/box_clip_op.cc @@ -35,7 +35,7 @@ class BoxClipOp : public framework::OperatorWithKernel { "The last dimension of InputBox must be 4"); 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, + PADDLE_ENFORCE_EQ(im_info_dims[1], 3, "The last dimension of ImInfo must be 3"); } ctx->ShareDim("InputBox", /*->*/ "OutputBox");