(三)TORCH.NN.FUNCTIONAL.INTERPOLATE使用说明


欢迎访问个人网络日志知行空间


1.TORCH.NN.FUNCTIONAL.INTERPOLATE

torch.nn.functional.interpolate(input, size=None, scale_factor=None, mode=‘nearest’, align_corners=None, recompute_scale_factor=None, antialias=False)

将输入input上采样或下采样到指定的size或缩放因子上scale_factor

resize使用的算法由mode参数决定,mode默认为nearest, 可选参数有nearest/linear(3D Inputs-only)/bilinear/bicubic(4D Inputs-only)/trilinear(5D Inputs-only)/area/nearest-exact

支持输入input为3/4/5维数据,输入数据维度的顺序为mini-batch x channels x [optional depth] x [optional height] x width

  • 参数
    • input:输入向量
    • size(int or Tuple[int] or Tuple[int, int] or Tuple[int, int, int])输出数据的空间尺度
    • mode (str) :采样算法
    • align_corners:输出的结果是否角对齐,对插值后的边的处理方式有所不同,参考一文看懂align_corners
    • recompute_scale_factor:是否重新计算scale_factor,默认None,选择特定的插值方式时可用

align_cornersFalseTrue时输出的对比:

(三)TORCH.NN.FUNCTIONAL.INTERPOLATE使用说明_第1张图片

2.示例

import torch
import torch.nn.functional as F

t = torch.tensor([[1,2,3],[4,5,6],[7,8,9]], dtype=torch.float).reshape(1,1,3,3)
print(t)
it0 = F.interpolate(t, mode = 'bilinear', scale_factor=2)
print(it0)

it1 = F.interpolate(t,  mode = 'bilinear', scale_factor=2, align_corners=True)
print(it1)

import cv2
t = t.squeeze().cpu().numpy().astype(np.float32)
img = cv2.resize(t, (6,6), interpolation=cv2.INTER_LINEAR)
print(f"opencv: \n{img}")

(三)TORCH.NN.FUNCTIONAL.INTERPOLATE使用说明_第2张图片

可以看到OpenCV中的resize函数与设置Pytorchalign_corners=False时的输出是相同的。

参考资料

  • 1.https://pytorch.org/docs/stable/generated/torch.nn.functional.interpolate.html
  • 2.https://zhuanlan.zhihu.com/p/87572724

欢迎访问个人网络日志知行空间


你可能感兴趣的