我正在使用torch.nn.functional.interpolate()
来调整图像大小。
首先,我使用transforms.ToTensor()
将图像转换为张量,张量的大小为(3, 252, 252),其中(252, 252)是导入图像的大小。我要做的是使用interpolate()
函数创建一个大小为(3, 504, 504)的张量。
我设置参数scale_factor=2
,但它返回了一个(3, 252, 504)的张量。然后我将其设置为scale_factor=(1,2,2)
,收到了维度冲突的错误信息:
size shape must match input shape. Input is 1D, size is 3
那么我应该如何设置参数才能接收到(3, 504, 504)的张量呢?