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 轉換