Opencv与wxpython的集成

Question

我想把我的网络摄像头的视频流整合到一个比高阶图形界面（highgui）更复杂的界面里。其实也没什么花哨的，就是几个按钮和其他一些东西，但这并没有那么简单。我找不到一个可以用来设计这个界面的基础示例。

我试着把这个代码转换成新的opencv接口，但结果很糟糕。我对opencv、numpy和图形界面设计都很陌生。有时候视频能正常播放，但大多数时候就卡在那里。我猜我可能犯了一个错误，就是在使用wx.BitmapFromBuffer(col, row, img)时，因为在旧版本中他们使用的是pil图像格式，而现在用的是numpy数组。所以在原始代码中，他们用的是pil的“imageData”函数，而我直接传递了numpy数组。

任何帮助都非常感谢。

这是我转换后的代码。

import wx
import cv2

class MyFrame(wx.Frame):
   def __init__(self, parent):
       wx.Frame.__init__(self, parent)
       self.displayPanel = wx.Panel(self)
       self.displayPanel.SetSize(wx.Size(800,640))

       self.cam = cv2.VideoCapture(1)
       self.cam.set(3, 640)
       self.cam.set(4, 480)
       ret, img = self.cam.read()

       cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
       row, col, x = img.shape
       self.SetSize((col,row))
       self.bmp = wx.BitmapFromBuffer(col, row, img)
       self.displayPanel.Bind(wx.EVT_PAINT, self.onPaint)

       self.playTimer = wx.Timer(self)
       self.Bind(wx.EVT_TIMER, self.onNextFrame)

       self.playTimer.Start(1000/15)

    def onPaint(self, evt):
        if self.bmp:
            dc = wx.BufferedPaintDC(self.displayPanel)
            self.PrepareDC(dc)
            dc.DrawBitmap(self.bmp, 0, 0, True)
        evt.Skip()

    def onNextFrame(self, evt):
        ret, img = self.cam.read()
        if ret == True:
            cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
            self.bmp.CopyFromBuffer(img)
            self.displayPanel.Refresh()
        evt.Skip()

if __name__=="__main__":
    app = wx.App()
    MyFrame(None).Show()
    app.MainLoop()

Answer 1

你应该在下面这一行加个注释

#self.PrepareDC(dc)

这个对我有效。

Answer 2

你需要设置面板的大小，以便显示捕获的图像。
我用了你的代码，并且我添加了“

self.SetSize(width,height)

这样可以

Answer 3

下面这个示例代码在我的OS X系统上运行得很好，但在不同平台上使用wx时，我遇到了一些小问题。代码几乎是一样的，唯一的不同是对cvtColor的结果进行了重新赋值，并且添加了一个wx.Panel的子类（这部分很重要）。

import wx
import cv, cv2

class ShowCapture(wx.Panel):
    def __init__(self, parent, capture, fps=15):
        wx.Panel.__init__(self, parent)

        self.capture = capture
        ret, frame = self.capture.read()

        height, width = frame.shape[:2]
        parent.SetSize((width, height))
        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)

        self.bmp = wx.BitmapFromBuffer(width, height, frame)

        self.timer = wx.Timer(self)
        self.timer.Start(1000./fps)

        self.Bind(wx.EVT_PAINT, self.OnPaint)
        self.Bind(wx.EVT_TIMER, self.NextFrame)


    def OnPaint(self, evt):
        dc = wx.BufferedPaintDC(self)
        dc.DrawBitmap(self.bmp, 0, 0)

    def NextFrame(self, event):
        ret, frame = self.capture.read()
        if ret:
            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
            self.bmp.CopyFromBuffer(frame)
            self.Refresh()


capture = cv2.VideoCapture(0)
capture.set(cv.CV_CAP_PROP_FRAME_WIDTH, 320)
capture.set(cv.CV_CAP_PROP_FRAME_HEIGHT, 240)

app = wx.App()
frame = wx.Frame(None)
cap = ShowCapture(frame, capture)
frame.Show()
app.MainLoop()