双线性插值计算手动实现以及原理-编程知识

双线性插值计算手动实现以及原理

代码
原理

代码

先贴代码吧，原理其实也比较简单，看代码基本也就理解了，时间太晚了，原理后续再补吧。

import torch
from torch.nn import functional as F
import numpy as np
from itertools import product
import warnings
warnings.filterwarnings('ignore')
def data_gen(in_wh,out_wh):in_w,in_h = in_whout_w,out_h = out_whinp = torch.range(0,in_h*in_w-1,1).reshape(1,1,in_h, in_w)new_h = torch.linspace(-1, 1, out_h).view(-1, 1).repeat(1, out_w)new_w = torch.linspace(-1, 1, out_w).repeat(out_h, 1)grid = torch.cat((new_w.unsqueeze(2), new_h.unsqueeze(2)), dim=2)#+2/5# print(grid.shape)grid = grid.unsqueeze(0).clip(-1,1)return inp,grid
def torch_interp(in_wh,out_wh):'''torch 插值'''inp,grid = data_gen(in_wh,out_wh)print(inp)outp = F.grid_sample(inp, grid=grid, align_corners=True)# print(outp)return outp
def my_interp(in_wh,out_wh):'''手动实现的插值计算'''inp,grid = data_gen(in_wh,out_wh)in_w,in_h = in_whp_h = 2/(in_h-1)#计算h方向上每一份的长度p_w = 2/(in_w-1)#计算w方向上每一份的长度p_ = np.array([p_w,p_h])#合并inp = inp[0,0,:,:]grid = grid[0]out = np.zeros((grid.shape[0],grid.shape[1]))#初始化输出模板for (_i,_j) in product(range(grid.shape[0]),range(grid.shape[1])):maps = (grid[_i,_j]).cpu().numpy()interp_start = (maps+1)//p_#计算插值的起始点offset = (maps+1)%p_#计算距离起始点的偏移量interp_start = interp_start.astype(np.int32)# print(interp_start)w, pix = [], []for j,i in [(0,0),(0,1),(1,0),(1,1)]:w.append(abs((i*p_[0]-offset[0])*(p_[1]*j-offset[1]))/(p_[0]*p_[1]))#双线性插值的面积计算interp_start_ = interp_start+np.array([i,j])#计算插值的四个点中的某一个interp_start_[0]=interp_start_[0].clip(0,in_w-1)#控制不超出原图像边界interp_start_[1]=interp_start_[1].clip(0,in_h-1)pix.append(inp[interp_start_[1]][interp_start_[0]])w = np.array(w)pix = np.array(pix[::-1])interp_val = round((w*pix).sum(),3)out[_i,_j]=interp_valreturn out
#原始输入
in_wh = 9, 9
#目标输出
out_wh =5, 2
a = torch_interp(in_wh,out_wh)
b = my_interp(in_wh,out_wh)
print(a,b)

结果：

tensor([[[[ 0.,  1.,  2.,  3.,  4.,  5.,  6.,  7.,  8.],[ 9., 10., 11., 12., 13., 14., 15., 16., 17.],[18., 19., 20., 21., 22., 23., 24., 25., 26.],[27., 28., 29., 30., 31., 32., 33., 34., 35.],[36., 37., 38., 39., 40., 41., 42., 43., 44.],[45., 46., 47., 48., 49., 50., 51., 52., 53.],[54., 55., 56., 57., 58., 59., 60., 61., 62.],[63., 64., 65., 66., 67., 68., 69., 70., 71.],[72., 73., 74., 75., 76., 77., 78., 79., 80.]]]])
(tensor([[[[ 0.,  2.,  4.,  6.,  8.],[72., 74., 76., 78., 80.]]]]),array([[ 0.,  2.,  4.,  6.,  8.],[72., 74., 76., 78., 80.]]))