mindspore.dataset.vision.py_transforms.Cutout

class mindspore.dataset.vision.py_transforms.Cutout(length, num_patches=1)[源代码]

随机去除输入numpy.ndarray图像上一定数量的正方形区域,将区域内像素值置为0。

请参阅论文 Improved Regularization of Convolutional Neural Networks with Cutout

参数:

  • length (int) - 去除正方形区域的边长。

  • num_patches (int,可选) - 去除区域的数量,默认值:1。

异常:

  • TypeError - 当 length 的类型不为int。

  • TypeError - 当 num_patches 的类型不为int。

  • ValueError - 当 length 小于等于0。

  • ValueError - 当 num_patches 小于等于0。

  • RuntimeError - 当输入图像的shape不为<H, W, C>。

支持平台:

CPU

样例:

>>> from mindspore.dataset.transforms.py_transforms import Compose
>>>
>>> transforms_list = Compose([py_vision.Decode(),
...                            py_vision.ToTensor(),
...                            py_vision.Cutout(80)])
>>> # apply the transform to dataset through map function
>>> image_folder_dataset = image_folder_dataset.map(operations=transforms_list,
...                                                 input_columns="image")