如何在将CV2 numpy数组转换为QImage时配置颜色？

Question

该程序基于pyqt和opencv。我计划在使用QT作为GUI时使用opencv读取和处理图像。

当我打开灰色图像时，结果没问题。但是在打开彩色图像时会改变图像的颜色。我想这是因为当我将numpy数组转换为OImage类型时我犯了错误，但我无法弄清楚如何纠正它。

# -*- coding: utf-8 -*-
from PyQt4.QtGui import *
from PyQt4.QtCore import *
import sys

import cv2
from cv2 import cv

QTextCodec.setCodecForTr(QTextCodec.codecForName("utf8"))

class MainWindow(QMainWindow):
    def __init__(self,parent=None):
        super(MainWindow,self).__init__(parent)
        self.setWindowTitle("Show Image with pyqt")

        self.imageLabel=QLabel()
        self.imageLabel.setSizePolicy(QSizePolicy.Ignored,QSizePolicy.Ignored)
        self.setCentralWidget(self.imageLabel)

        self.image=QImage()

        self.createActions()
        self.createMenus()

        self.cv_img = None

    def createActions(self):
        self.fileOpenAction=QAction(QIcon(":/fileopen.png"),self.tr("Open"),self)
        self.fileOpenAction.setShortcut("Ctrl+O")
        self.fileOpenAction.setStatusTip(self.tr("Open the file"))
        self.connect(self.fileOpenAction,SIGNAL("triggered()"),self.slotOpenFile)

        self.exitAction=QAction(QIcon(":/filequit.png"),self.tr("Quit"),self)
        self.exitAction.setShortcut("Ctrl+Q")
        self.setStatusTip(self.tr("Quit"))
        self.connect(self.exitAction,SIGNAL("triggered()"),self.close)

    def createMenus(self):
        fileMenu=self.menuBar().addMenu(self.tr("File"))
        fileMenu.addAction(self.fileOpenAction)
        fileMenu.addAction(self.exitAction)


    def slotOpenFile(self):
        fileName=QFileDialog.getOpenFileName(self,self.tr("Open a file"),\
            ".",self.tr("Image File(*.png *.jpg *.jpeg *.bmp)"))

        cvfilename=fileName.toLocal8Bit().data()

        if fileName.isEmpty()==False:

            cvfilename=fileName.toLocal8Bit().data() #convert Qstring to char*
            self.cv_img = cv2.imread(cvfilename) #read image with opencv
            cv2.imshow("Show Image with Opencv",self.cv_img) #show image with opencv, this is the right result

            self.image =QImage(self.cv_img.tostring(),\
                self.cv_img.shape[0],self.cv_img.shape[1],QImage.Format_RGB888) #convert numpy array to QImage
            self.imageLabel.setPixmap(QPixmap.fromImage(self.image))
            self.resize(self.image.width(),self.image.height())

app=QApplication(sys.argv)
main=MainWindow()
main.show()
app.exec_()

Answer 1

您需要将图像数据从BGR转换为RGB。您还需要交换宽度和高度（见下文） - 您的代码仅适用于具有相同宽度和高度的图像。

self.cv_img = cv2.imread(cvfilename)

if self.cv_img != None:
    # Notice the dimensions.
    height, width, bytesPerComponent = cv_img.shape
    bytesPerLine = bytesPerComponent * width;

    cv2.imshow("Show Image with Opencv", self.cv_img)

    # Convert to RGB for QImage.
    cv2.cvtColor(self.cv_img, cv.CV_BGR2RGB, self.cv_img)

    self.image = QImage(self.cv_img.data, width, height, bytesPerLine, QImage.Format_RGB888)

Answer 2

我知道这是一个很老的线程，但对我来说cv2.cvtColor工作得很慢。 我找到了另一个功能，可以帮助那些将来需要的人（让我永远找到这个补救措施）：

def cv2_to_qimage(cv_img):

    height, width, bytesPerComponent = cv_img.shape
    bgra = np.zeros([height, width, 4], dtype=np.uint8)
    bgra[:, :, 0:3] = cv_img
    return QtGui.QImage(bgra.data, width, height, QtGui.QImage.Format_RGB32)

希望你会发现它很有用

Answer 3

以下代码针对openCV3.x版本进行了 color-filter 更新。代码示例来自hluk。

import cv2
from cv2 import CV

...<snippet>...

self.cv_img = cv2.imread(cvfilename)

if self.cv_img != None:
    # Notice the dimensions.
    height, width, bytesPerComponent = cv_img.shape
    bytesPerLine = bytesPerComponent * width;

    cv2.imshow("Show Image with Opencv", self.cv_img)

    # Convert to RGB for QImage.
    cv2.cvtColor(self.cv_img, cv2.COlOR_BGR2RGB, self.cv_img)

    self.image = QImage(self.cv_img.data, width, height, bytesPerLine, QImage.Format_RGB888)