在多线程中使用Lock

时间:2018-05-24 06:36:58

标签: python multithreading opencv locking

我正在尝试使用muti(3)-threads并按下按钮同时保存来自三台摄像机的图像。

我选了一个code online并创建了三个对象实例。线程运行正常,窗口显示,可以保存图像。但我不确定我是否正在使用正确意义上的“锁定”(获取和发布)。总共有4把锁。

Current Code:
#!/usr/bin/env python

from threading import Thread, Lock
import cv2

class WebcamVideoStream :
    def __init__(self, src = 0, width = 1024, height = 768) :
        # width = 320, height = 240
        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.started = False
        self.read_lock = Lock()

    def start(self) :
        if self.started :
            print "already started!!"
            return None
        self.started = True
        self.thread = Thread(target=self.update, args=())
        self.thread.start()
        return self

    def update(self) :
        while self.started :
            #if self.stream.grab():
                #(grabbed, frame) = self.stream.retrieve()
            (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 stop(self) :
        self.started = False
        self.thread.join()

    def __exit__(self, exc_type, exc_value, traceback) :
        self.stream.release()

if __name__ == "__main__" :
    mainer_lock = Lock()
    vs = WebcamVideoStream(src = 0).start()
    vs2 = WebcamVideoStream(src = 1).start()
    vs3 = WebcamVideoStream(src = 2).start()

    while True :
        frame = vs.read()
        frame2 = vs2.read()
        frame3 = vs3.read()
        cv2.imshow('webcam', frame)
        cv2.imshow('webcam2', frame2)
        cv2.imshow('webcam3', frame3)

        if cv2.waitKey(1) == 99:
            mainer_lock.acquire()
            cv2.imwrite('cam_pics/frame.jpg', frame)
            cv2.imwrite('cam_pics/frame2.jpg', frame2)
            cv2.imwrite('cam_pics/frame3.jpg', frame3)
            mainer_lock.release()

    vs.stop()
    vs2.stop()
    vs3.stop()
    cv2.destroyAllWindows()

1 个答案:

答案 0 :(得分:1)

我认为你根本不需要锁定。

首先,mainer_lock毫无意义。我知道您希望使用它来保护frame[2, 3]不被vs[2, 3]更改。但这是不可能的,因为主线程中的frame[2, 3]是原始帧的副本(我想你正确使用copy)。 vs[2, 3]只会更改其实例中的框架,并且不会影响主线程中的框架。

其次,read_lock也毫无意义。你真的介意在阅读时改变框架吗?当您捕获真实帧时,后续帧几乎相同。所以,让我们说你想在1000毫秒捕获帧,但是因为线程竞争,你实际上在1002ms捕获帧。但它真的重要吗?据我所知,它没有。