capture.py 文件源码

def webcam(camera_id=0):
    # TODO update to support Python's 'with' construct
    camera = cv2.VideoCapture()
    camera.open(camera_id)

    # Use a smaller image size for faster processing
    camera.set(cv2.CAP_PROP_FRAME_WIDTH, 320.0)
    camera.set(cv2.CAP_PROP_FRAME_HEIGHT, 240.0)

    if not camera.isOpened():
        raise CameraInitializationError('Camera #{0} failed to open.'.format(camera_id))

    while camera.isOpened():
        success, frame = camera.read()
        if success:
            yield frame

    camera.release()