Question

我正在使用opencv尝试使用60fps的1920x1080高清摄像头拍摄视频流。问题不在于流媒体而是内存泄漏，我在流媒体的第一分钟内丢失了近6GB的内存。请帮我解决这个问题。

  from PySide.QtCore import *
  from PySide.QtGui import *
  import cv2
  import sys

  class MainApp(QWidget):

      def __init__(self):
        QWidget.__init__(self)
        self.video_size = QSize(1920, 1080)
        self.setup_ui()
        self.setup_camera()

    def setup_ui(self):
        """Initialize widgets.
        """
        self.image_label = QLabel()
        self.image_label.setFixedSize(self.video_size)

        self.quit_button = QPushButton("Quit")
        self.quit_button.clicked.connect(self.close)

        self.main_layout = QVBoxLayout()
        self.main_layout.addWidget(self.image_label)
        self.main_layout.addWidget(self.quit_button)

        self.setLayout(self.main_layout)

    def setup_camera(self):
        """Initialize camera.
        """
        self.capture = cv2.VideoCapture(0)
        self.capture.set(cv2.CAP_PROP_FRAME_WIDTH, self.video_size.width())
        self.capture.set(cv2.CAP_PROP_FRAME_HEIGHT, self.video_size.height())

        self.timer = QTimer()
        self.timer.timeout.connect(self.display_video_stream)
        self.timer.start(30)

    def display_video_stream(self):
        """Read frame from camera and repaint QLabel widget.
        """
        _, frame = self.capture.read()
        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
        image = QImage(frame, frame.shape[1], frame.shape[0],
                   frame.strides[0], QImage.Format_RGB888)
        self.image_label.setPixmap(QPixmap.fromImage(image))

if __name__ == "__main__":
    app = QApplication(sys.argv)
    win = MainApp()
    win.show()
    sys.exit(app.exec_())

修改

我发现内存泄漏的位置 self.timer.start（30）如果我增加时间调用，内存泄漏会更慢。有什么建议如何阻止内存泄漏？

Answer 1

如果有人有同样的问题，请导入qimage2ndarray这个库并添加对display_video_stream函数进行了以下更改

    def display_video_stream(self):
    """Read frame from camera and repaint QLabel widget.
    """

    _, frame = self.capture.read()

    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
    # frame = cv2.flip(frame, 1)
    image = qimage2ndarray.array2qimage(frame) #Solution for memory leak
    self.image_label.setPixmap(QPixmap.fromImage(image))

OpenCV：使用QPixmap

1 个答案: