我编写了一个OpenCV应用程序,它基本上从相机中抓取帧,进行一些图像处理并以两种编辑的变体显示图像。首先,我使用cv2.imshow()
来显示图像,但是当OpenCV(没有Qt支持的Build)无法提供现代GUI元素时,我决定使用PySide
作为我的GUI。
但是因为我在处理了大约830-850帧之后得到了这个错误(无论我使用什么计时器速率,或者我做了多少图像处理):
QImage: out of memory, returning null image
对于我在GUI中的两个图像视图,然后在每个循环中这个:
OpenCV Error: Unspecified error (The numpy array of typenum=2, ndims=3 can not be created) in NumpyAllocator::allocate, file ..\..\..\opencv-3.1.0\modules\python\src2\cv2.cpp, line 184
OpenCV Error: Insufficient memory (Failed to allocate 921600 bytes) in cv::OutOfMemoryError, file ..\..\..\opencv-3.1.0\modules\core\src\alloc.cpp, line 52
Traceback (most recent call last):
File "C:/myfile.py", line 140, in process_frame
img = QtGui.QImage(cv2.cvtColor(thresh_img, cv2.COLOR_RGB2BGR), self.width, self.height,
cv2.error: ..\..\..\opencv-3.1.0\modules\core\src\alloc.cpp:52: error: (-4) Failed to allocate 921600 bytes in function cv::OutOfMemoryError
这是我的代码的一部分(没有图像处理,但它也产生错误):
import cv2
import sys
from PySide import QtGui, QtCore
from threading import Thread
class MainWindow(QtGui.QMainWindow):
def __init__(self, cam=0, parent=None):
super(MainWindow, self).__init__(parent)
self.camera = Camera(cam).start()
self.title = "Cam %s" % cam
self.counter = 0
widget = QtGui.QWidget()
self.layout = QtGui.QBoxLayout(QtGui.QBoxLayout.LeftToRight)
self.video_frame = QtGui.QLabel()
self.thresh_frame = QtGui.QLabel()
self.layout.addWidget(self.video_frame)
self.layout.addWidget(self.thresh_frame)
self.layout.addStretch()
self.setCentralWidget(widget)
widget.setLayout(self.layout)
self.setMinimumSize(640, 480)
self._timer = QtCore.QTimer(self)
self._timer.timeout.connect(self.process_frame)
self._timer.start(20)
def process_frame(self):
self.counter += 1
print(self.counter)
self.frame = self.camera.read()
self.height, self.width = self.frame.shape[:2]
thresh_img = cv2.threshold(cv2.cvtColor(self.frame, cv2.COLOR_RGB2GRAY), 0, 255,
cv2.THRESH_BINARY | cv2.THRESH_OTSU)[1]
thresh_img = cv2.erode(thresh_img, None, iterations=2)
thresh_img = cv2.dilate(thresh_img, None, iterations=2)
thresh_img = cv2.cvtColor(thresh_img, cv2.COLOR_GRAY2RGB)
img = QtGui.QImage(cv2.cvtColor(self.frame, cv2.COLOR_RGB2BGR), self.width, self.height,
QtGui.QImage.Format_RGB888)
img = QtGui.QPixmap.fromImage(img)
self.video_frame.setPixmap(img)
img = QtGui.QImage(cv2.cvtColor(thresh_img, cv2.COLOR_RGB2BGR), self.width, self.height,
QtGui.QImage.Format_RGB888)
img = QtGui.QPixmap.fromImage(img)
self.thresh_frame.setPixmap(img)
def closeEvent(self, event):
self.camera.stop()
event.accept()
class Camera:
def __init__(self, src=0):
self.stream = cv2.VideoCapture(src)
(self.grabbed, self.frame) = self.stream.read()
self.stopped = False
def start(self):
Thread(target=self.update, args=()).start()
return self
def update(self):
while True:
if self.stopped:
return
(self.grabbed, self.frame) = self.stream.read()
def read(self):
return self.frame
def stop(self):
self.stopped = True
if __name__ == "__main__":
app = QtGui.QApplication(sys.argv)
window = MainWindow(0)
window.show()
sys.exit(app.exec_())
在Windows任务管理器中,我可以看到我的程序的RAM使用情况:
在崩溃时,该应用程序使用大约1.5 GB的RAM。我在gc
之后尝试使用gc.collect()
模块和del img
,但没有成功。
我还能做什么?
修改
线程Camera
类在这里无关紧要,没有它也会出现错误。
答案 0 :(得分:1)
它似乎是一个PySide特定的bug,使用PyQt将修复它。它甚至与OpenCV无关。现在看起来不会有使用PySide的解决方案...
答案 1 :(得分:1)
显然,这个错误是由于使用python 3.x时的垃圾收集问题
此处提供了使用ctypes的简单解决方法https://bugreports.qt.io/browse/PYSIDE-140和https://github.com/matplotlib/matplotlib/issues/4283#issuecomment-92773487
在最后一个链接上看到帖子" mfitzp于2015年4月24日发表评论"。这对我有用!
答案 2 :(得分:1)
就我而言,我使用QStackedWidget
在不同视图之间切换,QTimer
触发视图切换(在某些条件下)。
我使用functools.partial
将参数从一个视图传递到另一个视图,其中我确实传递了实例,例如PIL.Image
,QImage
,ImageQt
和{{1 }}。
那是错误的地方。当这些资源从一个视图传递到另一个视图时,垃圾收集器不能很好地清理它们。
对我有用。
将您计划从一个视图传递到另一个视图的所有变量声明为类属性(在QPixmap
的构造函数中)。将所有图片资源(例如QWidget
,PIL.Image
,QImage
和ImageQt
)存储在这些变量/属性中。
不要将任何参数(包含图像资源)从一个视图传递到另一个视图。
此外,如果您在OpenCV中使用PyQT,请注意OpenCV 2.9包含一个错误,导致内存泄漏。确保在花费数小时查找内存泄漏的来源之前升级到最新版本(最新的2.x或3.x)。
答案 3 :(得分:0)
查看我对PYSIDE-140的评论:https://bugreports.qt.io/browse/PYSIDE-140
避免将python对象与Qt对象混合。请尝试以下示例,避免使用修订修复。
stream = cv2.VideoCapture(0)
grabbed, frame = stream.read()
height, width = frame.shape[:2]
data = Qt.QByteArray(cv2.cvtColor(frame, cv2.COLOR_RGB2BGR).tostring())
qimg = Qt.QImage(data, width, height, Qt.QImage.Format_RGB888)
data.clear()
del data