python - 使用 PySide 将 QImage 转为 Numpy 数组

Question

我目前正在从 PyQt 切换到 PySide。

使用 PyQt，我使用我在 SO 上找到的代码将 QImage 转换为 Numpy.Array :

def convertQImageToMat(incomingImage):
    '''  Converts a QImage into an opencv MAT format  '''

    incomingImage = incomingImage.convertToFormat(4)

    width = incomingImage.width()
    height = incomingImage.height()

    ptr = incomingImage.bits()
    ptr.setsize(incomingImage.byteCount())
    arr = np.array(ptr).reshape(height, width, 4)  #  Copies the data
    return arr

但是 ptr.setsize(incomingImage.byteCount()) 不适用于 PySide，因为这是 void* support 的一部分PyQt 的。

我的问题是:如何使用 PySide 将 QImage 转换为 Numpy.Array。

编辑:

Version Info
> Windows 7 (64Bit)
> Python 2.7
> PySide Version 1.2.1
> Qt Version 4.8.5

Answer 1

对我来说，constBits() 的解决方案不起作用，但下面的解决方案起作用了:

def QImageToCvMat(incomingImage):
    '''  Converts a QImage into an opencv MAT format  '''

    incomingImage = incomingImage.convertToFormat(QtGui.QImage.Format.Format_RGBA8888)

    width = incomingImage.width()
    height = incomingImage.height()

    ptr = incomingImage.bits()
    ptr.setsize(height * width * 4)
    arr = np.frombuffer(ptr, np.uint8).reshape((height, width, 4))
    return arr