with（）语句从opencv中的VideoCapture读取？

Question

我喜欢使用with语句来访问文件和数据库连接，因为在发生错误或文件关闭时，它会自动为我断开连接。

f = open('file.txt', 'r')
for i in f():
   print(i)
f.close()

与

with open('file.txt', 'r') as f:
   for i in f:
       print(i)

在以下情况下，是否有等效的措辞用于读取相机缓冲区？

c = cv.VideoCapture(0)    
while(1):
    _,f = c.read()
    cv.imshow('e2',f)
    if cv.waitKey(5)==27:
        cv.waitKey()
        break
c.release()

我试过了：

c = cv.VideoCapture(0)    
while(1):
   with c.read() as _,f:
       cv.imshow('e2',f)
       if cv.waitKey(5)==27:
           cv.waitKey()
           break

---没有运气。拆解/释放似乎是另一种功能。这是成语吗？

Answer 1

我不知道 opencv，所以可能有更好的答案——但你总是可以通过定义和钩子来自己实现上下文管理器：__enter____exit__

class MyVideoCapture(cv.VideoCapture):
    def __enter__(self):
        return self
    def __exit__(self, *args):
        self.release()

用法如下：

with MyVideoCapture(0) as c:    
    while(True):
        _, f = c.read()
        cv.imshow('e2', f)
        if cv.waitKey(5) == 27:
            cv.waitKey()
            break

并且在您点击break语句后将释放资源。

根据您的评论，看起来opencv这里正在做一些时髦的事情。您还可以创建一个自定义类来包装 VideoCapture 实例：

class VideoCaptureWrapper(object):
    def __init__(self, *args, **kwargs):
        self.vid_stream = VideoCapture(*args, **kwargs)
    def __enter__(self):
        return self
    def __exit__(self, *args):
        self.vid_stream.release()

这里的用法是：

with VideoCaptureWrapper(0) as c:    
    while(True):
        _, f = c.vid_stream.read()
        cv.imshow('e2', f)
        if cv.waitKey(5) == 27:
            cv.waitKey()
            break

Answer 2

其它的办法：

from contextlib import contextmanager

@contextmanager
def VideoCapture(*args, **kwargs):
    cap = cv2.VideoCapture(*args, **kwargs)
    try:
        yield cap
    finally:
        cap.release()