python - 向量化迭代过程

Question

我用 python 编写了以下代码；它获取图像并将其从极坐标转换为笛卡尔坐标；效果是图像围绕一个点“展开”。

def unravel(img, origin):
  max_radius =  min(origin)

  out = np.zeros((max_radius,720,3), np.uint8)

  for phi in range(0,720):
    for r in range(0, max_radius):
      target = cmath.rect(r,math.radians(phi/2))
      out[(r,phi)] = img[(max_radius + target.real, max_radius+target.imag)]
  return out

这个算法真的很慢；我真的需要这个来操作实时视频。理想情况下，我希望能够将其“矩阵化”，以便使用 C 而不是 Python 循环执行底层计算。我在这方面不是特别有经验；什么是最好的方法？

Answer 1

这有点高级，但如果您希望能够使用矩阵变换流式传输视频，您将不得不亲自动手。这不是您可以使用当前方法合理完成的事情。

可能最方便的方法是使用 GStreamer使用自定义插件。有python bindings还有一些useful tutorials开始使用流水线。您可能想大量借用 geometrictransform插件(或者它们可能已经做了你想要的)。

您也可以尝试扩展您的问题并为 gstreamer 等添加标签。