如何使用Opencv从IP摄像机获取最新帧?

时间:2019-08-31 22:09:05

标签: python opencv raspberry-pi rtsp ip-camera

我需要为连接到我的rapiberry pi 3 b +的IP摄像机拍摄快照。我正在使用python3和opencv。相机中有一个计时器,所以我可以检查快照是否在正确的时间拍摄。

此脚本在装有Windows的PC上运行良好,但在树莓派中却无法运行。该脚本每秒钟拍摄一次快照,但是拍摄的帧不是正确的,而是旧的。

在树莓派中,我使用VLC和omxplayer来运行视频,并且运行流畅,因此我认为问题出在Opencv和我的代码中。我的印象是帧存储在缓冲区中,覆盆子太慢,无法实时从缓冲区中获取所有帧,因此随着时间的流逝,最后一个实际帧与所拍摄的帧之间会有更多的延迟。 / p>

import threading
import time
import cv2

cap = cv2.VideoCapture(‘rtsp://192.168.0.88’)
cap.set(cv2.CAP_PROP_BUFFERSIZE, 3)
counter = 0

while True:
   ret, frame = cap.read()
   if ret:
       cv2.imwrite(str(counter) + '.jpg', frame)
       counter = counter + 1
   time.sleep(1)

任何人都有一个想法,我怎么只能从相机中获取最新的照片? 我想说明一下,分辨率为1920x1080,视频格式为h264。此外,在实际应用中,我需要帧之间的时间为0.1秒。

1 个答案:

答案 0 :(得分:2)

enter image description here

这里是一个小部件,每隔x秒保存一次最新帧的屏幕截图。这个想法是创建另一个线程,仅用于获取帧,因为cv2.VideoCapture.read()是阻塞操作。通过将此操作放到仅专注于抓取帧的单独的专用线程中,我们可以确保我们拥有最新的帧而没有任何缓冲区。这将通过减少I / O延迟来提高性能,因为主线程不必等到新帧出现。我使用了自己的RTSP流链接,并每1秒保存一张屏幕截图。将其更改为您的RTSP链接,无论您想保存截图多长时间

from threading import Thread
import cv2
import time

class VideoScreenshot(object):
    def __init__(self, src=0):
        # Create a VideoCapture object
        self.capture = cv2.VideoCapture(src)

        # Take screenshot every x seconds
        self.screenshot_interval = 1

        # Default resolutions of the frame are obtained (system dependent)
        self.frame_width = int(self.capture.get(3))
        self.frame_height = int(self.capture.get(4))

        # Start the thread to read frames from the video stream
        self.thread = Thread(target=self.update, args=())
        self.thread.daemon = True
        self.thread.start()

    def update(self):
        # Read the next frame from the stream in a different thread
        while True:
            if self.capture.isOpened():
                (self.status, self.frame) = self.capture.read()

    def show_frame(self):
        # Display frames in main program
        if self.status:
            cv2.imshow('frame', self.frame)

        # Press Q on keyboard to stop recording
        key = cv2.waitKey(1)
        if key == ord('q'):
            self.capture.release()
            cv2.destroyAllWindows()
            exit(1)

    def save_frame(self):
        # Save obtained frame periodically
        self.frame_count = 0
        def save_frame_thread():
            while True:
                try:
                    cv2.imwrite('frame_{}.png'.format(self.frame_count), self.frame)
                    self.frame_count += 1
                    time.sleep(self.screenshot_interval)
                except AttributeError:
                    pass
        Thread(target=save_frame_thread, args=()).start()

if __name__ == '__main__':
    rtsp_stream_link = 'your stream link!'
    video_stream_widget = VideoScreenshot(rtsp_stream_link)
    video_stream_widget.save_frame()
    while True:
        try:
            video_stream_widget.show_frame()
        except AttributeError:
            pass