使用 wxPython 显示 OpenCV iplimage 数据结构

Question

这是我当前的代码（语言是Python）：

newFrameImage = cv.QueryFrame(webcam)
newFrameImageFile = cv.SaveImage("temp.jpg",newFrameImage)
wxImage = wx.Image("temp.jpg", wx.BITMAP_TYPE_ANY).ConvertToBitmap()
wx.StaticBitmap(self, -1, wxImage, (0,0), (wxImage.GetWidth(), wxImage.GetHeight()))

我试图在wxPython窗口中显示从网络摄像头捕获的iplimage。问题是我不想先将图像存储在硬盘上。有没有办法将 iplimage 转换为内存中的另一种图像格式？还有其他解决方案吗？

我在其他语言中找到了这个问题的一些“解决方案”，但我仍然遇到这个问题。

谢谢。

Answer 1

你要做的是：

frame = cv.QueryFrame(self.cam) # Get the frame from the camera
cv.CvtColor(frame, frame, cv.CV_BGR2RGB) # Color correction
                         # if you don't do this your image will be greenish
wxImage = wx.EmptyImage(frame.width, frame.height) # If your camera doesn't give 
                         # you the stream size, you might have to use (640, 480)
wxImage.SetData(frame.tostring()) # convert from cv.iplimage to wxImage
wx.StaticBitmap(self, -1, wxImage, (0,0), 
                (wxImage.GetWidth(), wxImage.GetHeight()))

我通过查看 Python OpenCV 食谱 以及 wxPython维基百科。

Answer 2

是的，这个问题很老了，但我和其他人一样来到这里寻找答案。在上述解决方案之后，我想我应该分享一个使用 cv2 和 numpy 图像的快速解决方案。

这是如何将 OpenCV2 中使用的 NumPy 数组样式图像转换为位图，然后您可以将其设置为 wxPython 中的显示元素（截至今天）：

import wx, cv2
import numpy as np

# Start with a numpy array style image I'll call "source"

# convert the colorspace to RGB from cv2 standard BGR, ensure input is uint8
img = cv2.cvtColor(np.uint8(source), cv2.cv.CV_BGR2RGB) 

# get the height and width of the source image for buffer construction
h, w = img.shape[:2]

# make a wx style bitmap using the buffer converter
wxbmp = wx.BitmapFromBuffer(w, h, img)

# Example of how to use this to set a static bitmap element called "bitmap_1"
self.bitmap_1.SetBitmap(wxbmp)

10 分钟前测试:)

这使用构建的在 wx 函数 BitmapFromBuffer 中，并利用 NumPy 缓冲区接口，以便所有我们要做的就是交换颜色以使它们按预期顺序排列。

Answer 3

你可以用 StringIO 来做，

stream = cStringIO.StringIO(data)
wxImage = wx.ImageFromStream(stream)

你可以在 \wx\lib\embeddedimage.py 中检查更多细节，

只是我的 2 美分。