python - 向量化 numpy 例程

Question

我编写了以下例程，它采用两条任意曲线并扭曲它们之间的空间，使其适合两条直线之间。对于循环，它按列处理它，因为 np.linspace 不对向量进行操作。有没有办法摆脱这个循环并立即解决整个问题？

def warp(img, iris_curve, pupil_curve):
  height, width = img.shape[:2]
  iris_height = np.uint8(np.max(np.int8(iris_curve) - pupil_curve))
  out = np.zeros((iris_height, width))

  for r in range(0,width):
    map_theta = np.linspace(pupil_curve[r], iris_curve[r], iris_height)
    map_theta = np.uint8(np.rint(map_theta))
    out[:, r] = img[map_theta, r]

  return np.uint8(out)

谢谢!

Answer 1

如果您查看source code np.linspace，您可以使用它作为代码矢量化的指南。然后你的循环将被替换为类似的内容:

steps = (iris_curve - pupil_curve) / iris_height
map_theta = np.arange(iris_height)[:, None] * steps + pupil_curve
map_theta = np.rint(map_theta).astype(np.uint8)
out = img[map_theta, np.arange(width)]

您可能必须转置输出，如果没有示例，很难调试此类代码。