快捷方式

CenterCrop

class torchvision.transforms.CenterCrop(size)[源代码]

将给定图像在中心裁剪。如果图像是 torch Tensor,则期望其形状为 […, H, W],其中 … 表示任意数量的前导维度。如果图像在任何边缘的尺寸小于输出尺寸,则图像将用 0 填充然后中心裁剪。

参数:

size (sequenceint) – 裁剪的期望输出尺寸。如果 size 是一个 int 而不是像 (h, w) 这样的序列,则会进行一个正方形裁剪 (size, size)。如果提供长度为 1 的序列,则将其解释为 (size[0], size[0])。

使用 CenterCrop 的示例

旋转边界框上的变换

旋转边界框上的变换

转换图示

转换图示
forward(img)[源代码]
参数:

img (PIL ImageTensor) – 要裁剪的图像。

返回:

裁剪后的图像。

返回类型:

PIL 图像或张量

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

查找开发资源并让您的问题得到解答

查看资源