python - torch.nn.functional.interpolate() : Parameters Settings

Question

我正在使用 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) 并收到如下尺寸冲突错误:

大小形状必须匹配输入形状。输入为1D，大小为3

那么应该怎么设置参数才能接收到(3, 504, 504)张量呢？

Answer 1

如果您使用的是 scale_factor，则需要提供一批图像而不是单个图像。因此，您需要使用 unsqueeze(0) 添加一批，然后将其交给 interpolate 函数，如下所示:

import torch
import torch.nn.functional as F

img = torch.randn(3, 252, 252)  # torch.Size([3, 252, 252])
img = img.unsqueeze(0)  # torch.Size([1, 3, 252, 252])

out = F.interpolate(img, scale_factor=(2, 2), mode='nearest')
print(out.size()) # torch.Size([1, 3, 504, 504])