class VideoFileStream:
def __init__(self, file, **kwargs):
self.vcap = cv2.VideoCapture(file)
self.stopped = False
self.fps = FPS(50, 100)
self.frame_singleton = SingletonBlockingQueue()
def start(self):
self.t = Thread(target=self.update, args=())
self.t.daemon = True
self.t.start()
return self
def update(self):
while (self.vcap.isOpened()):
ret, frame = self.vcap.read()
if not ret:
log.info('Reached the end of the video!')
self.stopped = True
else:
self.frame_singleton.enqueue(frame, wait=True)
self.fps.count()
if self.stopped:
self.vcap.release()
return
def read(self):
# return the frame most recently read
return self.frame_singleton.dequeue(notify=True)
def stop(self):
self.stopped = True
class RTMPVideoStream:
def __init__(self, rtmp_url, **kwargs):
log.info("rtmp capture init START")
self.vcap = cv2.VideoCapture(rtmp_url)
log.info("rtmp capture init END")
self.stopped = False
self.fps = FPS(50, 100)
self.frame_singleton = SingletonBlockingQueue()
def start(self):
self.t = Thread(target=self.update, args=())
self.t.daemon = True
self.t.start()
return self
def update(self):
total = 0
debug_frame = 100
while (self.vcap.isOpened()):
ret, frame = self.vcap.read()
if total % debug_frame == 0:
log.debug("rtmp capturing..")
self.frame_singleton.enqueue(frame)
self.fps.count()
total += 1
if self.stopped:
self.vcap.release()
return
def read(self):
# return the frame most recently read
return self.frame_singleton.dequeue()
def stop(self):
self.stopped = True
class PiVideoStream:
def __init__(self,
resolution=(320, 240),
framerate=32,
format='rgb',
**kwargs):
# initialize the camera
self.camera = PiCamera()
# set camera parameters
self.camera.resolution = resolution
self.camera.framerate = framerate
# set optional camera parameters (refer to PiCamera docs)
for (arg, value) in kwargs.items():
setattr(self.camera, arg, value)
# initialize the stream
self.rawCapture = PiRGBArray(self.camera, size=resolution)
self.stream = self.camera.capture_continuous(self.rawCapture,
format=format,
use_video_port=True)
# initialize the frame and the variable used to indicate
# if the thread should be stopped
self.stopped = False
self.fps = FPS(50, 100)
self.frame_singleton = SingletonBlockingQueue()
def start(self):
# start the thread to read frames from the video stream
self.t = Thread(target=self.update, args=())
self.t.daemon = True
self.t.start()
time.sleep(2.0)
return self
def update(self):
# keep looping infinitely until the thread is stopped
for f in self.stream:
# grab the frame from the stream and clear the stream in
# preparation for the next frame
self.frame_singleton.enqueue(f.array)
self.fps.count()
self.rawCapture.truncate(0)
# if the thread indicator variable is set, stop the thread
# and resource camera resources
if self.stopped:
self.stream.close()
self.rawCapture.close()
self.camera.close()
return
def read(self):
# return the frame most recently read
return self.frame_singleton.dequeue()
def stop(self):
# indicate that the thread should be stopped
self.stopped = True