diff --git a/mindspore/dataset/transforms/vision/c_transforms.py b/mindspore/dataset/transforms/vision/c_transforms.py index 1806d22446d..b46184fa16e 100644 --- a/mindspore/dataset/transforms/vision/c_transforms.py +++ b/mindspore/dataset/transforms/vision/c_transforms.py @@ -452,11 +452,11 @@ class Pad(cde.PadOp): class UniformAugment(cde.UniformAugOp): """ - Tensor operation to perform randomly selected augmentation + Tensor operation to perform randomly selected augmentation. Args: operations: list of C++ operations (python OPs are not accepted). - NumOps (int): number of OPs to be selected and applied. + num_ops (int): number of OPs to be selected and applied. Examples: >>> transforms_list = [c_transforms.RandomHorizontalFlip(), diff --git a/mindspore/dataset/transforms/vision/validators.py b/mindspore/dataset/transforms/vision/validators.py index 63cb7683c15..631d478d438 100644 --- a/mindspore/dataset/transforms/vision/validators.py +++ b/mindspore/dataset/transforms/vision/validators.py @@ -23,6 +23,7 @@ from .utils import Inter, Border from ...transforms.validators import check_pos_int32, check_pos_float32, check_value, check_uint8, FLOAT_MAX_INTEGER, \ check_bool, check_2tuple, check_range, check_list, check_type, check_positive, INT32_MAX + def check_inter_mode(mode): if not isinstance(mode, Inter): raise ValueError("Invalid interpolation mode.") @@ -838,7 +839,7 @@ def check_uniform_augmentation(method): if num_ops > len(operations): raise ValueError("num_ops is greater than operations list size") if not isinstance(operations, list): - raise ValueError("operations is not a python list") + raise TypeError("operations is not a python list") for op in operations: if not isinstance(op, TensorOp): raise ValueError("operations list only accepts C++ operations.")