我正在尝试设计将图像从笛卡尔坐标转换为极坐标的函数。要做到这一点,我需要使用插值,但我的代码运行速度极慢,瓶颈是以下功能:
def img2polar(img, center, final_radius, angle_resolution, Xcart, Ycart):
points = list(itertools.product(np.arange(0.5, img.shape[0]+0.5, 1.), np.arange(0.5, img.shape[1]+0.5, 1.)))
values = np.reshape(img, img.shape[0]*img.shape[1], 'C')
f = interpolate.griddata(points, values, (Xcart, Ycart), method='linear')
return f
有没有办法加速呢?
非常感谢!