mindspore.dataset.vision.py_transforms.FiveCrop

class mindspore.dataset.vision.py_transforms.FiveCrop(size)[source]

Crop the given image into one central crop and four corners.

Parameters

size (Union[int, Sequence[int, int]]) – The size of the cropped image. If int is provided, a square of size (size, size) will be cropped with this value. If Sequence[int, int] is provided, its two elements will be taken as the cropped height and width.

Raises
  • TypeError – If size is not of type int or Sequence[int, int].

  • ValueError – If size is not positive.

Supported Platforms:

CPU

Examples

>>> import numpy
>>> from mindspore.dataset.transforms.py_transforms import Compose
>>>
>>> transforms_list = Compose([py_vision.Decode(),
...                            py_vision.FiveCrop(size=200),
...                            # 4D stack of 5 images
...                            lambda *images: numpy.stack([py_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")