6

我只是想将来自我的网络摄像头的 opencv 视频流集成到一个比 highgui 所能提供的更复杂的 gui 中，没有什么特别的，只是几个按钮和其他东西，但事实证明它并不是那么微不足道。我找不到任何可以开始设计 gui 的基本示例。我尝试将此代码转换为新的 opencv 接口，结果很差。我是 opencv、numpy 和 gui 设计的新手。有时确实会流式传输视频，但大多数时候它只是挂在那里。我想我的一个错误可能是在 wx.BitmapFromBuffer(col, row, img) 中，因为在旧版本中他们使用 pil 图像格式，现在它使用 numpy 数组，所以在原始代码中使用了 pil 函数“imageData”，而不是正如我正在做的那样直接传递numpy数组。任何帮助它' color_channels_pic

这是我的代码转换。

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()
4

3 回答 3

12

下面的示例代码在 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()
于 2013-02-11T18:02:15.943 回答
0

您应该为以下行添加评论

#self.PrepareDC(dc)

它对我有用。

于 2020-06-19T09:25:23.913 回答
0

您必须设置面板的大小以显示捕获的图像。我使用了您的代码并添加了“

self.SetSize(width,height)

没关系

于 2018-03-06T02:33:52.810 回答