2017-02-28 50 views
0

我正在嘗試使用opencv從60fps的1920x1080高清攝像機進行視頻流。問題不在於流媒體,而在於內存泄露,在流媒體的第一分鐘內我損失了將近6GB的內存。請幫我解決這個問題。OpenCV:使用QPixmap時發生內存泄露

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

EDIT

我發現其中存儲器泄漏是self.timer.start(30)如果我增加時間呼叫時,內存泄漏慢。任何建議如何阻止內存泄漏?

回答

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)) 
相關問題