python - Pytorch 裁剪方法不填充输出

Question

我使用的是 torchvision 变换库中的裁剪方法。此处看到的方法描述（https://pytorch.org/vision/stable/transforms.html）表示，如果高度或宽度大于图像，它将用 0 填充。

但是，当我使用该功能时，它不是填充。在这里看到的输入

x = torch.rand(1,3,256,256)
y = TF.crop(img=x,top=0,left=0,height=300,width=300)
y.shape

结果是

torch.Size([1, 3, 256, 256])

我期望的输出是高度和宽度为 300 的东西。

标签： pythonpytorchpython-imaging-library

Answer 1

如果高度或宽度较大 .. 将用 0 填充 ..

没关系，但是如果你阅读完整的句子，它会说

如果图像尺寸小于输出 .. 用 0 填充然后裁剪

它会在填充后裁剪图像，这就是为什么你没有看到 300 的尺寸。