def MAT2QImage(self, im, copy=False):
""" convert CV MAT to QImage
see https://gist.github.com/smex/5287589 """
"""
Because OpenCV uses BGR order by default, you should first use
cvtColor(src, dst, CV_BGR2RGB)
to get an image layout that Qt understands
"""
im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB) # important!
gray_color_table = [qRgb(i, i, i) for i in range(256)]
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
