小编Mai*_*Son的帖子

torch.nn.fucntional.interpolate():参数设置

我用来torch.nn.functional.interpolate()调整图像大小。

首先,我将transforms.ToTensor()图像转换为张量,其大小为(3, 252, 252),(252, 252)是导入图像的大小。我想做的是使用interpolate()函数创建一个大小为 (3, 504, 504) 的张量。

我设置了 para 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) 张量呢?

python interpolation image-processing pytorch

3
推荐指数
1
解决办法
3587
查看次数

标签 统计

image-processing ×1

interpolation ×1

python ×1

pytorch ×1