From 5aad4bdfbf4d7b5afefe3c3efb66fab64a791d73 Mon Sep 17 00:00:00 2001 From: panfengfeng Date: Tue, 21 Jul 2020 12:02:58 +0800 Subject: [PATCH] change size --- .../ccsrc/minddata/dataset/kernels/image/center_crop_op.cc | 4 ++-- tests/ut/python/dataset/test_center_crop.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mindspore/ccsrc/minddata/dataset/kernels/image/center_crop_op.cc b/mindspore/ccsrc/minddata/dataset/kernels/image/center_crop_op.cc index 5bb9820c08..e33937b684 100644 --- a/mindspore/ccsrc/minddata/dataset/kernels/image/center_crop_op.cc +++ b/mindspore/ccsrc/minddata/dataset/kernels/image/center_crop_op.cc @@ -37,8 +37,8 @@ Status CenterCropOp::Compute(const std::shared_ptr &input, std::shared_p int32_t left = crop_wid_ - input->shape()[1]; std::shared_ptr pad_image; - CHECK_FAIL_RETURN_UNEXPECTED((top < input->shape()[0] * 10 && left < input->shape()[1] * 10), - "CenterCropOp padding size is too big, it's more than 10 times the original size."); + CHECK_FAIL_RETURN_UNEXPECTED((top < input->shape()[0] * 3 && left < input->shape()[1] * 3), + "CenterCropOp padding size is too big, it's more than 3 times the original size."); if (top > 0 && left > 0) { // padding only return Pad(input, output, top / 2 + top % 2, top / 2, left / 2 + left % 2, left / 2, BorderType::kConstant); diff --git a/tests/ut/python/dataset/test_center_crop.py b/tests/ut/python/dataset/test_center_crop.py index 746fbd4751..03b8079e1e 100644 --- a/tests/ut/python/dataset/test_center_crop.py +++ b/tests/ut/python/dataset/test_center_crop.py @@ -145,7 +145,7 @@ def test_center_crop_errors(): try: test_center_crop_op(16777216, 16777216) except RuntimeError as e: - assert "Unexpected error. CenterCropOp padding size is too big, it's more than 10 times the original size." in \ + assert "Unexpected error. CenterCropOp padding size is too big, it's more than 3 times the original size." in \ str(e)