python interpolate_Pytorch上下采样函数--interpolate用法

最近用到了上采样下采样操作，pytorch中使用interpolate可以很轻松的完成

举个例子：

这里注意上采样的时候mode默认是“nearest”，这里指定双线性插值“bilinear”

得到结果

补充知识：pytorch插值函数interpolate——图像上采样-下采样，scipy插值函数zoom

在训练过程中，需要对图像数据进行插值，如果此时数据是numpy数据，那么可以使用scipy中的zoom函数：

from scipy.ndimage.interpolation import zoom

中的zoom函数进行插值，

但是，如果此时的数据是tensor(张量)的时候，使用zoom函数的时候需要将tensor数据转为numpy，将GPU数据转换为CPU数据等，过程比较繁琐，可以使用pytorch自带的函数进行插值操作，interpolate函数有几个参数：size表示输出大小，scale_factor表示缩放倍数，mode表示插值方式，align_corners是bool类型，表示输入和输出中心是否对齐：

from torch.nn.functional import interpolate

以上这篇Pytorch上下采样函数--interpolate用法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持服务器之家。

原文链接：https://blog.csdn.net/zhaowangbo/article/details/89917884