执行此操作的一种方法可能是启动一个单独的线程,以正常速度读取视频文件并将每一帧填充到一个全局变量中。然后,主线程会在准备好时抓取该帧。
因此,如果我们使用仅显示时间和帧计数器的视频,我们可以在假相机更新帧时到处抓取帧:
#!/usr/bin/env python3
import cv2
import sys
import time
import logging
import numpy as np
import threading, queue
logging.basicConfig(level=logging.DEBUG, format='%(levelname)s %(message)s')
# This is shared between main and the FakeCamera
currentFrame = None
def FakeCamera(Q, filename):
"""Reads the video file at its natural rate, storing the frame in a global called 'currentFrame'"""
logging.debug(f'[FakeCamera] Generating video stream from {filename}')
# Open video
video = cv2.VideoCapture(filename)
if (video.isOpened()== False):
logging.critical(f'[FakeCamera] Unable to open video {filename}')
Q.put('ERROR')
return
# Get height, width and framerate so we know how often to read a frame
h = int(video.get(cv2.CAP_PROP_FRAME_HEIGHT))
w = int(video.get(cv2.CAP_PROP_FRAME_WIDTH))
fps = video.get(cv2.CAP_PROP_FPS)
logging.debug(f'[FakeCamera] h={h}, w={w}, fps={fps}')
# Initialise currentFrame
global currentFrame
currentFrame = np.zeros((h,w,3), dtype=np.uint8)
# Signal main that we are ready
Q.put('OK')
while True:
ret,frame = video.read()
if ret == False:
break
# Store video frame where main can access it
currentFrame[:] = frame[:]
# Try and read at appropriate rate
time.sleep(1.0/fps)
logging.debug('[FakeCamera] Ending')
Q.put('DONE')
if __name__ == '__main__':
# Create a queue for synchronising and communicating with our fake camera
Q = queue.Queue()
# Create a fake camera thread that reads the video in "real-time"
fc = threading.Thread(target=FakeCamera, args=(Q,'video.mov'))
fc.start()
# Wait for fake camera to intialise
logging.debug(f'[main] Waiting for camera to power up and initialise')
msg = Q.get()
if msg != 'OK':
sys.exit()
# Main processing loop should go here - we'll just grab a couple frames at different times
cv2.imshow('Video',currentFrame)
res = cv2.waitKey(2000)
cv2.imshow('Video',currentFrame)
res = cv2.waitKey(5000)
cv2.imshow('Video',currentFrame)
res = cv2.waitKey(2000)
# Wait for buddy to finish
fc.join()
样本输出
DEBUG [FakeCamera] Generating video stream from video.mov
DEBUG [main] Waiting for camera to power up and initialise
DEBUG [FakeCamera] h=240, w=320, fps=25.0
DEBUG [FakeCamera] Ending
关键词:Python、图像处理、OpenCV、多线程、多线程、多线程、多线程、视频、假相机。