将 cython 内存视图传递给 OpenCV 函数

Question

我正在使用 cython memoryviews 来引用一些灰度图像。我已经在我编写的一些图像处理代码中成功地使用了它。现在，我需要使用一些 OpenCV 函数。 不幸的是，我发现我无法将内存视图作为图像参数传递给 OpenCV 函数。代码可以编译，但运行时会在 OpenCV 函数调用处停止，并显示“TypeError: is not a numpy array”

我可以使用 np.asarray(my_memoryview) 将 memoryview 转换回 numpy 数组。这可行，但它复制数据并且速度很慢。

在 memoryview 文档中，他们谈到强制转换为 numpy http://docs.cython.org/src/userguide/memoryviews.html#coercion-to-numpy 似乎我应该能够在不复制内存的情况下将 memoryview 强制为 numpy 数组。但是，如果我写：

im = np.asarray(<np.uint8_t[:, :]> my_memoryview)

导致编译错误：“Can only create cython.array from pointer or array”

任何有关如何将 memoryview 传递给 OpenCV 函数，或如何在不复制数据的情况下将 memoryview 强制转换为 numpy 数组的帮助将不胜感激！

Answer 1

Numpy/OpenCV 不使用内存视图，但需要 legacy precursor。您可以创建一个包装类：

from cython.view cimport memoryview

cdef extern from "Python.h":
    object PyLong_FromVoidPtr(void *p)

cdef class OpenCVMemoryView:
    cdef object arr
    cdef object underlying_object
    def __init__(OpenCVMemoryView self, np.uint8_t[:, :] my_memoryview):
        self.underlying_object = my_memoryview # prevents GC of my_memoryview
        cdef memoryview my_memoryview_c = my_memoryview
        self.arr = dict(version=3,
            typestr='<u1', #typestr=np.uint8,
            data=(PyLong_FromVoidPtr(<void*>my_memoryview_c.view.buf), False),
            strides=my_memoryview.strides,
            shape=my_memoryview.shape)
    def __array_interface__(self):
        return self.arr

Cython memoryview object 具有返回元组的属性，如 __array_interface__ 所要求的。

如果这不是更快，我会推断您的解决方案已经不是复制数据。