p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

repair parameter check problem in random_resize_crop and random_crop

This commit is contained in:
ms_yan 2020-03-31 20:13:49 +08:00 committed by chang zherui
parent 51feea03a4
commit 7a16dfe2bf
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/dataset/transforms/vision/validators.py
View File

@ -41,7 +41,7 @@ def check_crop_size(size):
else: else:
raise TypeError("Size should be a single integer or a list/tuple (h, w) of length 2.") raise TypeError("Size should be a single integer or a list/tuple (h, w) of length 2.")
for value in size: for value in size:
check_value(value, (1, INT32_MAX)) check_pos_int32(value)
return size return size
@ -239,6 +239,7 @@ def check_random_resize_crop(method):
kwargs["scale"] = scale kwargs["scale"] = scale
if ratio is not None: if ratio is not None:
check_range(ratio, [0, FLOAT_MAX_INTEGER]) check_range(ratio, [0, FLOAT_MAX_INTEGER])
check_positive(ratio[0])
kwargs["ratio"] = ratio kwargs["ratio"] = ratio
if interpolation is not None: if interpolation is not None:
check_inter_mode(interpolation) check_inter_mode(interpolation)