pj3fizlut
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

check input param `degree` in (min, max) format in random_rotation and other op

This commit is contained in:
ms_yan 2020-07-23 17:47:00 +08:00
parent 9b8d38eab4
commit d7c01210cc
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -94,7 +94,11 @@ def check_degrees(degrees):
if isinstance(degrees, numbers.Number):
check_value(degrees, (0, float("inf")), "degrees")
elif isinstance(degrees, (list, tuple)):
if len(degrees) != 2:
if len(degrees) == 2:
type_check_list(degrees, (numbers.Number,), "degrees")
if degrees[0] > degrees[1]:
raise ValueError("degrees should be in (min,max) format. Got (max,min).")
else:
raise TypeError("If degrees is a sequence, the length must be 2.")
@ -106,6 +110,8 @@ def check_random_color_adjust_param(value, input_name, center=1, bound=(0, FLOAT
raise ValueError("The input value of {} cannot be negative.".format(input_name))
elif isinstance(value, (list, tuple)) and len(value) == 2:
check_range(value, bound)
if value[0] > value[1]:
raise ValueError("value should be in (min,max) format. Got (max,min).")
def check_erasing_value(value):
@ -168,10 +174,14 @@ def check_random_resize_crop(method):
type_check(scale, (tuple,), "scale")
type_check_list(scale, (float, int), "scale")
check_range(scale, [0, FLOAT_MAX_INTEGER])
if scale[0] > scale[1]:
raise ValueError("scale should be in (min,max) format. Got (max,min).")
if ratio is not None:
type_check(ratio, (tuple,), "ratio")
type_check_list(ratio, (float, int), "ratio")
check_range(ratio, [0, FLOAT_MAX_INTEGER])
if ratio[0] > ratio[1]:
raise ValueError("ratio should be in (min,max) format. Got (max,min).")
if interpolation is not None:
type_check(interpolation, (Inter,), "interpolation")
if max_attempts is not None: