Shortcuts

center_crop

torchvision.transforms.functional.center_crop(img: Tensor, output_size: List[int]) Tensor[source]

在中心裁剪给定的图像。 如果图像是torch Tensor,则期望其具有[…, H, W]形状,其中…表示任意数量的前导维度。 如果图像的任何边缘尺寸小于输出尺寸,则图像用0填充,然后进行中心裁剪。

Parameters:
  • img (PIL ImageTensor) – 要裁剪的图像。

  • output_size (sequenceint) – 裁剪框的(高度,宽度)。如果是整数或包含单个整数的序列,则用于两个方向。

Returns:

裁剪后的图像。

Return type:

PIL 图像或张量

使用 center_crop 的示例:

变换的图示

Illustration of transforms