center_crop¶
- torchvision.transforms.functional.center_crop(img: Tensor, output_size: List[int]) Tensor [source]¶
将给定的图像从中心位置裁剪。如果图像是 torch Tensor,它应具有 […, H, W] 的形状,其中 … 表示任意数量的前导维度。如果图像沿任一边缘的大小小于输出大小,图像将用 0 填充,然后进行中心裁剪。
- 参数:
img (PIL Image 或 Tensor) – 要裁剪的图像。
output_size (sequence 或 int) – 裁剪框的 (高度, 宽度)。如果为 int 或只包含一个 int 的序列,则对两个方向使用相同的值。
- 返回:
裁剪后的图像。
- 返回类型:
PIL Image or Tensor
使用
center_crop
的示例