mindspore.dataset.vision.FiveCrop
- class mindspore.dataset.vision.FiveCrop(size)[源代码]
在输入PIL图像的中心与四个角处分别裁剪指定尺寸大小的子图。
参数:
size (Union[int, Sequence[int, int]]) - 裁剪子图的尺寸大小。若输入int,则以该值为边长裁剪( size , size )尺寸大小的子图;若输入Sequence[int, int],则以2个元素分别为高和宽裁剪子图。
异常:
TypeError - 当 size 的类型不为int或Sequence[int, int]。
ValueError - 当 size 不为正数。
- 支持平台:
CPU
样例:
>>> import numpy >>> from mindspore.dataset.transforms import Compose >>> >>> transforms_list = Compose([vision.Decode(to_pil=True), ... vision.FiveCrop(size=200), ... # 4D stack of 5 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")