mindspore.dataset.vision.c_transforms.UniformAugment

class mindspore.dataset.vision.c_transforms.UniformAugment(transforms, num_ops=2)[源代码]

对输入图像执行随机选取的数据增强操作。

参数:

  • transforms (TensorOperation) - 对给定图像随机选择的边界框区域应用 C++ 变换处理。(不接受Python操作)。

  • num_ops (int, 可选) - 要选择和执行的操作的数量,默认值:2。

异常:

  • TypeError - 如果 transform 不是 mindspore.dataset.vision.c_transforms 模块中的图像变换处理。

  • TypeError - 当 num_ops 不是int类型。

  • ValueError - 当 num_ops 不为正数。

支持平台:

CPU

样例:

>>> import mindspore.dataset.vision.py_transforms as py_vision
>>> transforms_list = [c_vision.RandomHorizontalFlip(),
...                    c_vision.RandomVerticalFlip(),
...                    c_vision.RandomColorAdjust(),
...                    c_vision.RandomRotation(degrees=45)]
>>> uni_aug_op = c_vision.UniformAugment(transforms=transforms_list, num_ops=2)
>>> transforms_all = [c_vision.Decode(), c_vision.Resize(size=[224, 224]),
...                   uni_aug_op]
>>> image_folder_dataset_1 = image_folder_dataset.map(operations=transforms_all,
...                                                   input_columns="image",
...                                                   num_parallel_workers=1)