python opencv 入门 加载视频 （2）

内容来自OpenCV-Python Tutorials 自己翻译整理

目标：

1. 读取视频、显示视频、保存视频
2. 使用摄像头拍摄并显示
3. 学习cv2.VideoCapture(), cv2.VideoWriter()函数

使用摄像头拍摄视频：
（我的用的是笔记本电脑上面的摄像头）
OpenCV 提供了一个非常简单的接口。我们把视频转换成黑白格式，然后显示出来。

首先要创建一个VideoCapture对象。参数可以是设备标示（device index）或是一个视频文件的名称。设备标示是指定使用哪个摄像头的编号。通常一个摄像头会连接上。所以只需要传递参数0或者-1。可以选择第二个摄像头通过传递参数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() 函数返回一个布尔类型。如果读的视频帧是正确的，返回True，所以可以通过返回值检查。

有时，cap可能没有初始化摄像头，可以检查是否初始化或者使用cap.isOpened() 函数，如果返回True则继续，否则使用cap.open()

可以使用cap.get(proid)函数来设置视频的属性，proid是0到18的数字每个数组表示一个视频的属性，具体见Property Identifier
其中一些属性可以是用cap.set(proid,value)方法修改

例如，使用cap.get(3)和cap.get(4)获得每一帧的宽度和高度，获得的结果是640和480，可以把它修改成320×240
ret = cap.set(3,320) ret = cap.set(4,240)

#!/usr/bin/env python
# coding=utf-8

import cv2
import numpy as np
from matplotlib import pyplot as plt

cap = cv2.VideoCapture(0)
ret = cap.set(3,320)
ret = cap.set(4,240)
while(True):
    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()

（不要把set写到循环里，否则会很卡顿）

播放视频文件：
只需要把VideoCapture里面的摄像头标示改成文件名即可，如果视频很小，播放的太快，可以使用参数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()

存储视频：
要把摄像机拍摄到的视频按帧存起来可以是用cv2.imwrite()函数

首先建立一个VideoWriter对象，然后指定要保存的文件名，例如output.avi。接下来使用FourCC，接下来传递每秒的帧数和帧的大小参数，最后是isColor标记，如果该标记为真，将得到彩色的图片。

FourCC 是用于指定视频编解码器 4 字节代码。代码可以在fourcc.org上找到，该代码是平台相关的。(本人机器为ubuntu 14.04)

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()

得到的视频是掉转过来的