forked from pradana.aumars/videocr
16 lines
382 B
Python
16 lines
382 B
Python
|
import cv2
|
||
|
|
||
|
|
||
|
class Capture:
|
||
|
def __init__(self, video_path):
|
||
|
self.path = video_path
|
||
|
|
||
|
def __enter__(self):
|
||
|
self.cap = cv2.VideoCapture(self.path)
|
||
|
if not self.cap.isOpened():
|
||
|
raise IOError('Can not open video {}.'.format(self.path))
|
||
|
return self.cap
|
||
|
|
||
|
def __exit__(self, exc_type, exc_value, traceback):
|
||
|
self.cap.release()
|