python 将 IplImage 转换为 Qimage

Question

我正在尝试使用 openCV 处理一些视频，然后将其放入 pyqt Qimage...

我看到了一些例子，但它们都是用 C++ 编写的，我只能理解 python，

谁能帮帮我...谢谢

Answer 1

您可以使用以下代码将 numpy 数组转换为 QImage:

from PyQt4.QtGui import QImage, qRgb
import numpy as np

class NotImplementedException:
    pass

gray_color_table = [qRgb(i, i, i) for i in range(256)]

def toQImage(im, copy=False):
    if im is None:
        return QImage()

    if im.dtype == np.uint8:
        if len(im.shape) == 2:
            qim = QImage(im.data, im.shape[1], im.shape[0], im.strides[0], QImage.Format_Indexed8)
            qim.setColorTable(gray_color_table)
            return qim.copy() if copy else qim

        elif len(im.shape) == 3:
            if im.shape[2] == 3:
                qim = QImage(im.data, im.shape[1], im.shape[0], im.strides[0], QImage.Format_RGB888);
                return qim.copy() if copy else qim
            elif im.shape[2] == 4:
                qim = QImage(im.data, im.shape[1], im.shape[0], im.strides[0], QImage.Format_ARGB32);
                return qim.copy() if copy else qim

    raise NotImplementedException

然后在调用 toQImage()

之前将 OpenCV 的 CvMat 转换为 numpy 数组

arr = numpy.asarray(mat)
qim = toQImage(arr)

另见 http://opencv.willowgarage.com/documentation/python/cookbook.html用于 OpenCV 的 CvMat 和 numpy 数组之间的转换。