CenterCrop

class torchvision.transforms.v2.CenterCrop(size: Union[int, Sequence[int]])

從中心裁剪輸入。

如果輸入是 torch.Tensor 或 TVTensor (例如 Image, Video, BoundingBoxes 等)，它可以有任意數量的前導批次維度。例如，影像可以具有 [..., C, H, W] 形狀。邊界框可以具有 [..., 4] 形狀。

如果影像在任何邊緣上的尺寸小於輸出尺寸，影像將用 0 填充，然後從中心裁剪。

引數:

size (sequence 或 int) – 期望的裁剪輸出尺寸。如果 size 是一個整數而不是像 (h, w) 這樣的序列，則進行一個正方形裁剪 (size, size)。如果提供一個長度為 1 的序列，則將其解釋為 (size[0], size[0])。

使用 CenterCrop 的示例

轉換的圖示

轉換的圖示

transform(inpt: Any, params: Dict[str, Any]) → Any

用於覆蓋以實現自定義轉換的方法。

請參閱 如何編寫自己的 v2 轉換