Video_capture/Simple_video_register/backend/app/video_stream.py

13 lines
278 B
Python

import cv2
class VideoStream:
def __init__(self, rtsp_url):
self.rtsp_url = rtsp_url
self.stream = cv2.VideoCapture(rtsp_url)
def get_frame(self):
ret, frame = self.stream.read()
if not ret:
return None
return frame