shenwei41
5 years ago
1 changed files with 4 additions and 2 deletions
Unified View
Diff Options
+ 4
- 2
mindspore/dataset/transforms/vision/validators.py
View File
| @@ -519,8 +519,8 @@ def check_random_affine(method): | |||||
| check_degrees(degrees) | check_degrees(degrees) | ||||
| if translate is not None: | if translate is not None: | ||||
| if type_check(translate, (list, tuple), "translate"): | |||||
| type_check_list(translate, (int, float), "translate") | |||||
| type_check(translate, (list, tuple), "translate") | |||||
| type_check_list(translate, (int, float), "translate") | |||||
| if len(translate) != 2: | if len(translate) != 2: | ||||
| raise TypeError("translate should be a list or tuple of length 2.") | raise TypeError("translate should be a list or tuple of length 2.") | ||||
| for i, t in enumerate(translate): | for i, t in enumerate(translate): | ||||
| @@ -528,6 +528,7 @@ def check_random_affine(method): | |||||
| if scale is not None: | if scale is not None: | ||||
| type_check(scale, (tuple, list), "scale") | type_check(scale, (tuple, list), "scale") | ||||
| type_check_list(scale, (int, float), "scale") | |||||
| if len(scale) == 2: | if len(scale) == 2: | ||||
| for i, s in enumerate(scale): | for i, s in enumerate(scale): | ||||
| check_positive(s, "scale[{}]".format(i)) | check_positive(s, "scale[{}]".format(i)) | ||||
| @@ -541,6 +542,7 @@ def check_random_affine(method): | |||||
| if isinstance(shear, numbers.Number): | if isinstance(shear, numbers.Number): | ||||
| check_positive(shear, "shear") | check_positive(shear, "shear") | ||||
| else: | else: | ||||
| type_check_list(shear, (int, float), "shear") | |||||
| if len(shear) not in (2, 4): | if len(shear) not in (2, 4): | ||||
| raise TypeError("shear must be of length 2 or 4.") | raise TypeError("shear must be of length 2 or 4.") | ||||
| if len(shear) == 2 and shear[0] > shear[1]: | if len(shear) == 2 and shear[0] > shear[1]: | ||||