cv2解码多线程实现

白衣祭酒

已于 2022-09-20 11:49:43 修改

阅读量365

点赞数

分类专栏： cv 文章标签： python 开发语言 opencv numpy

于 2022-09-20 10:22:10 首次发布

本文链接：https://blog.csdn.net/qq_41650747/article/details/126947947

版权

cv 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

看到了一个使用线程解码的类，使用了线程中的 Event 作为开关控制，记录一下。

可以解决计算过程中读取帧越来越慢的问题，看起来比队列方式更省资源

链接：https://github.com/SakshayMahna/Computer-Vision-on-Humans/blob/main/PoseEstimation/camera.py

from threading import Thread, Lock
from datetime import datetime
import time
import cv2

time_cycle = 80

class CameraThread(Thread):
    def __init__(self, kill_event, src = 0, width = 320, height = 240):
        self.kill_event = kill_event
        
        self.stream = cv2.VideoCapture(src)
        self.stream.set(cv2.CAP_PROP_FRAME_WIDTH, width)
        self.stream.set(cv2.CAP_PROP_FRAME_HEIGHT, height)

        (self.grabbed, self.frame) = self.stream.read()
        self.read_lock = Lock()

        Thread.__init__(self, args = kill_event)

    def update(self):
        (grabbed, frame) = self.stream.read()
        self.read_lock.acquire()
        self.grabbed, self.frame = grabbed, frame
        self.read_lock.release()

    def read(self):
        self.read_lock.acquire()
        frame = self.frame.copy()
        self.read_lock.release()
        return frame

    def run(self):
        while not self.kill_event.is_set():
            start_time = datetime.now()

            self.update()

            finish_time = datetime.now()
            dt = finish_time - start_time
            ms = (dt.days * 24 * 60 * 60 + dt.seconds) * 1000 + dt.microseconds / 1000.0
            if ms < time_cycle:
                time.sleep((time_cycle - ms) / 1000.0)
 



if __name__ == "__main__":
	from threading import Event
	import cv2
	width = 640
	height = 480
	kill_event = Event()
	cam = CameraThread(kill_event, height = height, width = width)
    cam.start()

	kill_event.set()#关闭解码