center_crop¶
- torchvision.transforms.functional.center_crop(img: Tensor, output_size: list[int]) Tensor[原始碼]¶
在影像中心進行裁剪。如果影像是 torch Tensor,它應該具有 […, H, W] 形狀,其中 … 表示任意數量的前導維度。如果影像尺寸在任何邊緣小於輸出尺寸,則影像將用 0 填充然後進行中心裁剪。
- 引數:
img (PIL Image 或 Tensor) – 要裁剪的影像。
output_size (序列或int) – 裁剪框的 (高, 寬)。如果為 int 或只包含一個 int 的序列,則兩者都用於兩個方向。
- 返回:
裁剪後的影像。
- 返回型別:
PIL 影像或張量
使用
center_crop的示例