功能描述
算子功能
Resampling算子可以对输入tensor进行放缩。存在最邻近(nearest neighbor)插值算法和线性(linear)插值算法两种实现方式。
算子公式
- 最邻近算法数学公式为:dst(n,c,oh,ow) = src(n,c,ih,iw),其中:
- ih=[(oh+0.5)/Fh−0.5]
- iw=[(ow+0.5)/Fw−0.5]
- 双线性采样的数学公式为:dst(n,c,oh,ow) = src(n,c,ih0,iw0)*(1 - Wih)*(1 - Wiw) +src(n,c,ih1,iw0)*Wih*(1 - Wiw) + src(n,c,ih0,iw1) * (1 - Wih)*Wiw +src(n,c,ih1,iw1)*Wih*Wiw,其中:
- ih0=⌊oh+0.5Fh−0.5⌋
- ih1=⌈oh+0.5Fh−0.5⌉
- iw0=⌊ow+0.5Fw−0.5⌋
- iw1=⌈ow+0.5Fw−0.5⌉
- Wih=oh+0.5Fh−0.5−ih0
- Wiw=ow+0.5Fw−0.5−iw0
表1 公式参数说明 参数
参数说明
src
源输入Tensor。
dst
目标Tensor。
n
待采样的维度1。
c
待采样的维度2。
ih
输入Height。
iw
输入Width。
oh
输出Height。
ow
输出Width。
父主题: Resampling