Opencv视频处理

目标

• 学习阅读视频，显示视频和保存视频。
• 学会从摄像机中捕捉并显示出来。
• 您将学习以下功能： cv2.VideoCapture(), cv2.VideoWriter()

从摄像机中获取视频

通常情况下，我们必须用相机捕捉实时流。OpenCV提供了一个非常简单的接口。让我们从摄像机中获取一个视频(我正在使用我的笔记本电脑内置的网络摄像头)，将它转换成灰度视频并显示出来。只是一个简单的开始任务。

要捕获视频，您需要创建一个视频拍摄对象。它的参数可以是设备索引，也可以是视频文件的名称。设备索引只是指定哪个摄像机的数字。正常情况下，会有一个摄像头连接起来(就像我的情况一样)。所以我只需传递0(或-1)。像在Ubuntu系统中，可以通过在终端中输入命令 ls/dev 查看摄像头的设备号。您可以通过传递1来选择第二个相机，以此类推。在那之后，你可以一帧地捕捉。但最后别忘了释放抓捕。

import numpy as np
import cv2

cap = cv2.VideoCapture(0)

while(True):
    # Capture frame-by-frame
    ret, frame = cap.read()

    # Our operations on the frame come here
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    # Display the resulting frame
    cv2.imshow('frame',gray)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# When everything done, release the capture
cap.release()
cv2.destroyAllWindows()

cap.read()返回bool值(真/假)。如果frame被正确地读取，它将是True。因此，您可以通过检查这个返回值来检查视频是否读取成功。有时候，cap可能没有初始化捕获。在这种情况下，此代码显示错误。您可以通过方法cap.isOpened()检查它是否初始化。

从文件中播放视频

它与从摄像机拍摄一样，只需更改摄像机索引与视频文件名。同时，在显示frame时，请使用适当的时间cv2.waitKey()。如果它太少，视频将是非常快，如果它太高，视频将是缓慢的。25毫秒在正常情况下是可以的。

import numpy as np
import cv2

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

while(cap.isOpened()):
    ret, frame = cap.read()

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

    cv2.imshow('frame',gray)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

注：确保安装了适当的ffmpeg或gStreer版本。有时候，使用视频捕获是一个头疼的问题，主要是因为错误安装了ffmpeg/gStreer。

保存视频

我们捕获一个视频，逐帧处理它，我们想保存这个视频。对于图像，它非常简单，只需使用cv2.imwrite()。视频的话，这里需要做更多的工作。

这一次我们创建了一个VideoWriter对象。我们应该指定输出文件名(例如：output.avi)。然后，我们应该指定FourCC 编码(详见下一段)。然后应该传递每秒帧数(Fps)和帧大小。最后一个是isColor flag。如果它是True，编码器期望颜色帧，否则它将为灰度帧工作。

FourCC 用于指定视频编解码器的4字节代码。

• 在Fedora：DIVX, XVID, MJPG, X264, WMV1, WMV2。(XVID更可取，MJPG的结果是高尺寸的视频，X264提供非常小尺寸的视频)
• 在Windows中：XVID(需要测试和添加更多内容)

下面的代码作用为从相机捕捉并且在垂直方向翻转每一帧，并保存它。

import numpy as np
import cv2

cap = cv2.VideoCapture(0)

# Define the codec and create VideoWriter object
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('output.avi',fourcc, 20.0, (640,480))

while(cap.isOpened()):
    ret, frame = cap.read()
    if ret==True:
        frame = cv2.flip(frame,0)

        # write the flipped frame
        out.write(frame)

        cv2.imshow('frame',frame)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break
    else:
        break

# Release everything if job is finished
cap.release()
out.release()
cv2.destroyAllWindows()