实现视频的读、显、存

  • 学会从摄像头获取并显示视频
  • 学习相关函数：cv2.VideoCapture()，cv2.VideoWrite()

一、用摄像头捕获视频

我们经常需要使用摄像头捕获实时图像。OpenCV 为这中应用提供了一个非常简单的接口cv2.VideoCapture(0)。让我们使用摄像头来捕获一段视频，并把它转换成灰度视频显示出来。

为了获取视频：

  1. 创建一个 VideoCapture 对象。参数可以是设备的索引号，或者是一个视频文件。设备索引号就是在指定要使用的摄像头。一般的笔记本电脑都有内置摄像头。所以参数就是 0。你可以通过设置成 1 或者其他的来选择别的摄像头。
  2. 一帧一帧的捕获视频。
  3. 停止捕获视频。

cap.read() 返回一个布尔值（True/False）。如果帧读取的是正确的，就是 True。所以最后你可以通过检查他的返回值来查看视频文件是否已经到了结尾。
有时 cap 可能不能成功的初始化摄像头设备。这种情况下上面的代码会报错。你可以使用 cap.isOpened()，来检查是否成功初始化了。如果返回值是True，那就没有问题。否则就要使用函数 cap.open()。

可以使用函数 cap.get(propId) 来获得视频的一些参数信息。这里propId 可以是 0 到 18 之间的任何整数。每一个数代表视频的一个属性，见下表其中的一些值可以使用cap.set(propId,value) 来修改，value 就是你想要设置成的新值。

例如，可以使用 cap.get(3) 和 cap.get(4) 来查看每一帧的宽和高。默认情况下得到的值是 640X480。但是同样可以使用 ret=cap.set(3,320)和 ret=cap.set(4,240) 来把宽和高改成 320X240。

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(10) & 0xFF == 27:
    	break

# 停止捕获，释放对象
cap.release()
cv2.destroyAllWindows()

二、读取视频文件

  • 与从摄像头中捕获一样，只需要把设备索引号改成视频文件的名字。
  • 在播放每一帧时，使用 cv2.waiKey() 设置适当的持续时间。如果设置的太低视频就会播放的非常快，如果设置的太高就会播放的很慢（你可以使用这种方法控制视频的播放速度）。通常情况下 25 毫秒就可以。
import cv2

cap = cv2.VideoCapture('test.mp4')

# 检查是否打开正确
if cap.isOpened(): 
    oepn, frame = cap.read()
else:
    open = False
    
# 循环读取    
while open:
    ret, frame = cap.read()
    if frame is None:
        break
    if ret == True:
        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
        cv2.imshow('result', gray)
        if cv2.waitKey(10) & 0xFF == 27:
            break
                
cap.release()
cv2.destroyAllWindows()
import cv2

cap = cv2.VideoCapture('test.mp4')

# 循环读取    
while (cap.isOpened()):
    ret, frame = cap.read()
    if frame is None:
        break
    if ret == True:
        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
        cv2.imshow('result', gray)
        if cv2.waitKey(10) & 0xFF == 27:
            break
                
cap.release()
cv2.destroyAllWindows()

三、保存视频

当我们需要保存通过摄像头捕获的视频，或者在我们对视频处理后想保存处理效果时，就可以参考以下代码。

在我们捕获视频，并对每一帧都进行加工之后我们想要保存这个视频。对于视频来说,要创建一个 VideoWriter 的对象。确定一个输出文件的名字。

import cv2
 
# 打开视频
cap = cv2.VideoCapture("test.mp4")

# 视频的宽度
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
# 视频的高度
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))


# 视频的帧率  视频的编码  定义视频输出
fps = cap.get(cv2.CAP_PROP_FPS)
fourcc = int(cap.get(cv2.CAP_PROP_FOURCC))
writer = cv2.VideoWriter("out.mp4", fourcc, fps, (width, height))


while (cap.isOpened()):
    ret, frame = cap.read()
    if frame is None:
        break
    if ret:
        # 显示
        cv2.imshow('视频', frame)
        # 写入新的视频
        writer.write(frame)
        if cv2.waitKey(10) & 0xFF == 27:
            break

cap.release()
writer.release()
cv2.destroyAllWindows()