|
|
|
@@ -37,8 +37,8 @@ Status CenterCropOp::Compute(const std::shared_ptr<Tensor> &input, std::shared_p |
|
|
|
int32_t left = crop_wid_ - input->shape()[1]; |
|
|
|
std::shared_ptr<Tensor> 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); |
|
|
|
|