scipy curve_fit与整数参数

Question

我正在尝试使用scipy的curve_fit将两个灰度图像之间的x和y偏移量计算为2-D numpy ndarrays。 但是，这要求我提供的curve_fit函数的参数需要整数参数。

因此，我有一个带有三个参数的函数：展平的数组和沿两个轴的偏移量。 它可以正确完成偏移量，并为所有整数偏移量返回重新展平的数组，但是curve_fit始终返回p0。

是否可以使curve_fit与整数参数一起使用？

Answer 1

这里一种可能的方法可能是定义两个浮点偏移量的函数，该函数在整数点之间执行插值：

def func_orig(xint,yint):
    ....

def func_new(x,y):
    x1 = int(x)
    y1 = int(y)
    x2 = x1 + 1
    y2 = y1 + 1
    val = func_orig(x1,y1) * (x2-x)*(y2-y)+
          func_orig(x1,y2) * (x2-x)*(y1-y)+
          func_orig(x2,y1) * (x1-x)*(y2-y)+
          func_orig(x2,y2) * (x1-x)*(y1-y)
    val = val / (x2-x1)/(y2-y1)
    return val