当我需要用Python内插数据时,我通常使用C中数值食谱中的双线性和双三次例程的实现(W.H. Press,1992)。主要的原因是我没有得到我所期望的。由于我现在需要更多的灵活性(例如不规则网格),所以我想使用griddata函数。
下面的脚本可以说明我的问题:
from matplotlib import pyplot as plt
from scipy.interpolate import griddata
from interp import * # My own interpolation functions (from Press. 1992)
# Create test
我正在尝试理解如何使用Pytorch的网格示例函数。我知道我们传入了一个B*通道*H*W和一个UV像素的流场B*H*W*2
但它似乎不起作用。
import numpy as np
import torch
import torch.nn.functional as F
from torch import nn
from torch.autograd import Variable
import time
import cv2
rgbimg = np.ones((100,100,3)).astype(np.float32)*0.5
rgbimg[0:50,0:50,:] = 0
rgbi