mindspore.dataset.vision.TenCrop

class mindspore.dataset.vision.TenCrop(size, use_vertical_flip=False)[源代码]

在输入PIL图像的中心与四个角处分别裁剪指定尺寸大小的子图,并将其翻转图一并返回。

参数:
  • size (Union[int, Sequence[int, int]]) - 裁剪子图的尺寸大小。若输入int,则以该值为边长裁剪( size , size )尺寸大小的子图;若输入Sequence[int, int],则以2个元素分别为高和宽裁剪子图。

  • use_vertical_flip (bool,可选) - 若为True,将对子图进行垂直翻转;否则进行水平翻转。默认值:False。

异常:
  • TypeError - 当 size 的类型不为int或Sequence[int, int]。

  • TypeError - 当 use_vertical_flip 的类型不为bool。

  • ValueError - 当 size 不为正数。

支持平台:

CPU

样例:

>>> import numpy
>>> from mindspore.dataset.transforms import Compose
>>>
>>> transforms_list = Compose([vision.Decode(to_pil=True),
...                            vision.TenCrop(size=200),
...                            # 4D stack of 10 images
...                            lambda *images: numpy.stack([vision.ToTensor()(image) for image in images])])
>>> # apply the transform to dataset through map function
>>> image_folder_dataset = image_folder_dataset.map(operations=transforms_list,
...                                                 input_columns="image")