CenterCrop¶
- class torchvision.transforms.CenterCrop(size)[源代码]¶
在中心裁剪给定图像。如果图像是 torch Tensor,则期望其形状为 […, H, W],其中 … 表示任意数量的前导维度。如果图像沿任何边缘的大小小于输出大小,则用 0 填充图像,然后进行中心裁剪。
- 参数:
size (sequence 或 int) – 裁剪的期望输出尺寸。如果 size 是一个整数而不是像 (h, w) 这样的序列,则会进行一个正方形裁剪 (size, size)。如果提供一个长度为 1 的序列,它将被解释为 (size[0], size[0])。
使用
CenterCrop
的示例