forked from mindspore-Ecosystem/mindspore
repair parameter problem in Uniform augmentation
This commit is contained in:
parent
a510ecc8ff
commit
d6f27f5a4a
|
@ -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(),
|
||||
|
|
|
@ -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.")
|
||||
|
|
Loading…
Reference in New Issue