OpenCV函数不适用于在PyQt GUI窗口中转换的OpenCV窗口

时间:2015-03-01 18:13:42

标签: python qt opencv image-processing pyqt

我正在使用OpenCV进行图像处理并将图像嵌入到PyQt GUI窗口中。我独立使用了OpenCV代码,工作正常。我还可以在OpenCV中创建以下窗口和对象,并通过cv2.cvtColor成功转换其他颜色空间,然后转换为QPixmap并显示它们。

当我尝试对OpenCV图像做任何事情而不是转换颜色空间时,会出现问题。我尝试过简单的OpenCV操作,例如cv.Smooth,甚至使用cv.CreateImage创建一个阈值图像,但这些都不起作用。我不确定是否需要很长时间来执行这些操作并且QTGui正在超时。 请帮帮我,我不明白什么是错的。我在下面提供了我的代码片段:

import cv2    
import cv2.cv as cv
import numpy as np
import from PyQt4 import QtGui, QtCore

class Video():
    def __init__(self,capture):
        self.capture = capture
        self.capture.open(1)
        self.currentFrame=np.array([])
        self.capture.set(3,800)
        self.capture.set(4,600)
        self.capture.set(5,30)

    def processImage(self):
        try:
            while(True):
                ret, readFrame=self.capture.read() # Grab next frame
                if(ret==True):
                    print "processImage grabbed a frame"

                    #readFrame=cv2.cvtColor(readFrame,cv2.COLOR_BGR2HSV) < -- This works fine
                    #readFrame=cv2.cvtColor(readFrame,cv2.COLOR_HSV2RGB) < -- This works fine
                    cv.Smooth(readFrame,readFrame,cv.CV_BLUR,3) < -- Any functions like this cause problems
                    print 'It made it this far' # < -- This never gets printed because cv.Smooth doesn't properly execute
        except TypeError:
            print "ERROR: processImage couldn't get a frame"

        def convertFrame(self):
    """     converts frame to format suitable for QtGui            """
        try:
            height,width=self.currentFrame.shape[:2]
            img=QtGui.QImage(self.currentFrame,width,height,QtGui.QImage.Format_RGB888)
            img=QtGui.QPixmap.fromImage(img)
            self.previousFrame = self.currentFrame
            return img
        except:
            return None

我甚至尝试使用cv.Smooth然后使用cv2.cvtColor将RGB转换为QT Gui并且无关紧要;一旦它击中cv.Smooth或除了cvtColor之外的任何其他OpenCV函数,它就不会超过它。

以下是GUI的代码:

class Gui(QtGui.QMainWindow):
    def __init__(self,parent=None):
        QtGui.QWidget.__init__(self,parent)
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        self.video = Video(cv2.VideoCapture()) # Which device to capture frames from
        self._timer = QtCore.QTimer(self)
        self._timer.timeout.connect(self.play)
        self._timer.start(27)
        self.update()

    def play(self):
        try:
            self.video.processImage()
            self.ui.videoFrame.setPixmap(self.video.convertFrame())
            self.ui.videoFrame.setScaledContents(True)
        except TypeError:
            print "No frame"

def main():
    app = QtGui.QApplication(sys.argv)
    ex = Gui()
    ex.show()
    sys.exit(app.exec_())

if __name__ == '__main__':
    main()

这可能与调用processImage()的Qt GUI有关,并且在上一个函数返回之前继续下一次调用processImage()需要很长时间来处理图像并超时吗?

请帮助!!!

1 个答案:

答案 0 :(得分:1)

真的没有人试图回答这个问题吗?好吧,如果您遇到这种情况需要弄明白,这就是解决问题的方法:

  1. 始终使用cv2类而不是旧的cv类。

  2. 在我的代码中,我试图使用cv.QueryFrame,它返回一个cvMat,而pyqt由于某种原因不喜欢这种数据类型。

  3. 您需要将所有旧的cv函数转换为新的cv2函数。调用capture.read()而不是cv.QueryFrame(capture)会返回一个VideoCapture对象,这是pyqt可以处理的数据类型。

  4. 找出相应的功能使用cv2是一件苦差事,但一旦你开始工作,这是值得的。