test=develop

Author: zhupengyang

lite/operators/roi_perspective_transform_op.cc

@@ -32,7 +32,7 @@ bool RoiPerspectiveTransformOp::CheckShape() const {
       << "The format of input tensor must be NCHW. But received input dims is: "
       << x_dims;
   auto rois_dims = param_.rois->dims();
-  CHECK_EQ(rois_dims.size(), 1UL)
+  CHECK_EQ(rois_dims.size(), 2UL)
       << "ROIs should be a 2-D LoDTensor of shape (num_rois, 8) given as [[x0, "
          "y0, x1, y1, x2, y2, x3, y3], ...]. But received rois dims: "
       << rois_dims;