OpenCV 入门从0开始第二天笔记

学习目标：读取视频，显示视频，保存视频

从相机中读取视频，通常情况下，我们用摄像机捕捉一个视频，提供一个简单的画面，让我们从摄像头捕捉一段视频，将其转换为灰度图显示

捕捉视频使用的是 VedioCaputre 对象，她的对象可以是设备索引或视频文件的名称，设备索引就是指定哪个摄像头，一个摄像头会被链接，简单传0或-1，之后你可以逐帧捕获视频，不要忘记释放俘虏

import numpy as np

import cv2 as cv

cap = cv.VideoCapture(0)

if not cap.isOpened():

    print("Cannot open camera")

    exit()

while True:

    # 逐帧捕获

    ret, frame = cap.read()

    # 如果正确读取帧，ret为True

    if not ret:

        print("Can't receive frame (stream end?). Exiting ...")

        break
    # 我们在框架上的操作到这里

    gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)

    # 显示结果帧e

    cv.imshow('frame', gray)

    if cv.waitKey(1) == ord('q'):

        break

# 完成所有操作后，释放捕获器

cap.release()

cv.destroyAllWindows()

cap.read() 如果正确读取了帧，它将为True 因此可以检查返回值来检查视频的末尾

有时 cap还没开始捕获 可以通过cap.isOpen()的方法来检查是否初始化了，如果为True ,那么就是用 cap.open()打开它

高度和宽度 可以通过cap.get(cv.CAP_DROP_FRAME_WIDTH)和cap.get(cv.CAP_DROP_FRAME_WIDTH)   来检查高度和宽度 默认的为 640x480 

修改高度和宽度使用 ret=cap.set(cv.CAP_DROP_FRAME_WIDTH,320),cap.set(cv.CAP_DROP_FRAME_WIDTH,240)

从文件中播放视频

import numpy as np

import cv2 as cv

cap = cv.VideoCapture('vtest.avi')

while cap.isOpened():

    ret, frame = cap.read()

    # 如果正确读取帧，ret为True

    if not ret:

        print("Can't receive frame (stream end?). Exiting ...")

        break

    gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)

    cv.imshow('frame', gray)

    if cv.waitKey(1) == ord('q'):

        break

cap.release()

cv.destroyAllWindows()

保存视频

所以我们捕捉一个视频，一帧一帧地处理，我们想要保存这个视频。对于图像，它非常简单，只需使用 cv.imwrite()。这里还需要做一些工作。

这次我们创建一个 VideoWriter 对象。我们应该指定输出文件名(例如: output.avi)。然后我们应该指定 FourCC 代码(详见下一段)。然后传递帧率的数量和帧大小。最后一个是颜色标志。如果为True，编码器期望颜色帧，否则它与灰度帧一起工作。

• 在Fedora中：DIVX，XVID，MJPG，X264，WMV1，WMV2。（最好使用XVID。MJPG会生成大尺寸的视频。X264会生成非常小的尺寸的视频）
• 在Windows中：DIVX（尚待测试和添加）
• 在OSX中：MJPG（.mp4），DIVX（.avi），X264（.mkv）

import numpy as np

import cv2 as cv

cap = cv.VideoCapture(0)

# 定义编解码器并创建VideoWriter对象

fourcc = cv.VideoWriter_fourcc(*'XVID')

out = cv.VideoWriter('output.avi', fourcc, 20.0, (640,  480))

while cap.isOpened():

    ret, frame = cap.read()

    if not ret:

        print("Can't receive frame (stream end?). Exiting ...")

        break

    frame = cv.flip(frame, 0)

    # 写翻转的框架
    
    out.write(frame)

    cv.imshow('frame', frame)

    if cv.waitKey(1) == ord('q'):

        break

# 完成工作后释放所有内容

cap.release()

out.release()

cv.destroyAllWindows()