1. 首页
  2. 编程语言
  3. Python
  4. Pytorch上下采样函数--interpolate用法

Pytorch上下采样函数--interpolate用法

上传者: 2022-03-05 16:41:55上传 PDF文件 76.69 KB 热度 14次

最近用到了上采样下采样操作,pytorch中使用interpolate可以很轻松的完成举个例子:这里注意上采样的时候mode默认是“nearest”,这里指定双线性插值“bilinear”得到结果在训练过程中,需要对图像数据进行插值,如果此时数据是numpy数据,那么可以使用scipy中的zoom函数:。中的zoom函数进行插值,但是,如果此时的数据是tensor(张量)的时候,使用zoom函数的时候需要将tensor数据转为numpy,将GPU数据转换为CPU数据等,过程比较繁琐,可以使用pytorch自带的函数进行插值操作,interpolate函数有几个参数:size表示输出大小,scale_factor表示缩放倍数,mode表示插值方式,align_corners是bool类型,表示输入和输出中心是否对齐:

下载地址
用户评论