关于pytorch的upsample_bilinear2d的align_corners和half

代码位置在：aten/src/Aten/native/cpu/UpSampleMoreKernel.cpp，关于bilinear的坐标计算过程：

template <typename scalar_t>
static inline scalar_t area_pixel_compute_source_index(scalar_t scale,int64_t dst_index,bool align_corners,bool cubic) {if (align_corners) {return scale * dst_index;            //half pixel为false} else {scalar_t src_idx = scale * (dst_index + 0.5) - 0.5;    //half pixel为true// [Note] Follow Opencv resize logic:// We allow negative src_idx here and later will use//   dx = src_idx - floorf(src_idx)// to compute the "distance"(which affects weights).// For linear modes, weight distribution doesn't matter// for negative indices as they use 2 pixels to interpolate.// For example, [-1, 0], they both use pixel 0 value so it// doesn't affect if we bound the src_idx to 0 or not.// TODO: Our current linear mode impls use unbound indices// where we should and then remove this cubic flag.// This matters in cubic mode, as we might need [-1, 0, 1, 2]// to interpolate and the weights can be affected.return (!cubic && src_idx < 0) ? scalar_t(0) : src_idx;}
}

关于nearest2d，可以看出计算中half_pixel应该是false

static inline int64_t nearest_neighbor_compute_source_index(const float scale,int64_t dst_index,int64_t input_size) {const int64_t src_index =std::min(static_cast<int64_t>(floorf(dst_index * scale)), input_size - 1);return src_index;
}

关于pytorch的upsample_bilinear2d的align_corners和half_pixel等参数确定相关推荐

PyTorch | （4）神经网络模型搭建和参数优化
PyTorch | (1)初识PyTorch PyTorch | (2)PyTorch 入门-张量 PyTorch | (3)Tensor及其基本操作 PyTorch | (4)神经网络模型搭建和参数 ...
pytorch卷积操作nn.Conv中的groups参数用法解释
MobileNetV1<MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications> ...
pytorch：nn.ReLU(inplace=True) 中的参数 inplane 的作用
pytorch 的激活函数nn.Relu() 有一个参数:inplace,其作用是:该nn.Relu() 函数计算得到的输出是否更新传入的输出. 正常情况下,一个卷积操作,首先经过卷积层,接着是正则化 ...
Pytorch 中的数据类型 torch.utils.data.DataLoader 参数详解
DataLoader是PyTorch中的一种数据类型,它定义了如何读取数据方式.详情也可参考本博主的另一篇关于torch.utils.data.DataLoader(https://blog.csdn ...
Pytorch:使用官网提供数据集的相关参数设置，以CIFAR10为例进行说明
文章目录前言一.Dataset 定义-组成分类二.获取数据集 1.参数说明 2.相关Demo 前言本文记录笔者关于Dataset的相关学习记录,以Pytorch官网文档为主进行学习一.Da ...
pytorch加载预训练加载部分参数
最简单的: state_dict = torch.load(weight_path) self.load_state_dict(state_dict,strict=False) 加载cpu: m ...
pytorch加载之前训练模型中的部分参数以及冻结部分参数（实测，自己实际项目代码中的）
我的需求是,由于我在不停的尝试各种模型,导致模型木块一直会变.如果每次重复重新开始训练要花费大把时间. 我之前运行的模型 ResNet -> ...
【pytorch】Module.parameters()函数实现与网络参数管理
我们知道可以通过Module.parameters()获取网络的参数,那这个是如何实现的呢?我先直接看看函数的代码实现: def parameters(self):r""" ...
pytorch系列 -- 9 pytorch nn.init 中实现的初始化函数 uniform, normal, const, Xavier, He initialization...
本文内容: 1. Xavier 初始化 2. nn.init 中各种初始化函数 3. He 初始化 torch.init https://pytorch.org/docs/stable/nn.html ...

关于pytorch的upsample_bilinear2d的align_corners和half_pixel等参数确定

关于pytorch的upsample_bilinear2d的align_corners和half_pixel等参数确定相关推荐

最新文章

热门文章