python - 如何调整 PyTorch 张量的大小？

Question

我有一个大小为 (5, 1, 44, 44)(批次、 channel 、高度、宽度)的 PyTorch 张量，我想将其“调整大小”为 (5, 1, 224, 224)我怎样才能做到这一点？我应该使用哪些功能？

Answer 1

看起来您正在寻找 interpolate (nn.functional 中的一个函数):

import torch.nn.functional as nnf

x = torch.rand(5, 1, 44, 44)
out = nnf.interpolate(x, size=(224, 224), mode='bicubic', align_corners=False)

如果你真的关心插值的准确性，你应该看看 ResizeRight :一个 pytorch/numpy 包，它可以在调整图像大小时准确地处理各种“边缘情况”。这在直接合并不同尺度的特征时会产生影响:不准确的插值可能会导致未对齐。