From 624ab97de6f28444b7bff27f743deddcc0a80a46 Mon Sep 17 00:00:00 2001 From: ms_yan <6576637+ms_yan@user.noreply.gitee.com> Date: Tue, 31 Mar 2020 20:13:49 +0800 Subject: [PATCH] repair parameter check problem in random_resize_crop and random_crop --- mindspore/dataset/transforms/vision/validators.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mindspore/dataset/transforms/vision/validators.py b/mindspore/dataset/transforms/vision/validators.py index 17f50e068d..443232d780 100644 --- a/mindspore/dataset/transforms/vision/validators.py +++ b/mindspore/dataset/transforms/vision/validators.py @@ -41,7 +41,7 @@ def check_crop_size(size): else: raise TypeError("Size should be a single integer or a list/tuple (h, w) of length 2.") for value in size: - check_value(value, (1, INT32_MAX)) + check_pos_int32(value) return size @@ -239,6 +239,7 @@ def check_random_resize_crop(method): kwargs["scale"] = scale if ratio is not None: check_range(ratio, [0, FLOAT_MAX_INTEGER]) + check_positive(ratio[0]) kwargs["ratio"] = ratio if interpolation is not None: check_inter_mode(interpolation)