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的示例