06-python opencv 读取视频文件并保存

概述

本节实现的是使用内建摄像头捕获视频，并保存视频。

• 创建摄像头对象
• 逐帧显示实现视频播放
• 保存摄像头的每一帧图像

实现过程

引用

不再赘述，代码如下。

import cv2  
import numpy

创建摄像头对象

使用opencv自带的VideoCapture()函数定义摄像头对象，其参数0表示第一个摄像头，一般就是笔记本的内建摄像头。

cap = cv2.VideoCapture(0)

定义写入视频格式和写入对象

利用cv2.cv.FOURCC(*’XVID’)定义视频格式，然后创建视频写入对象。

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

逐帧显示视频并写入

在while循环中，利用视频对象的read()函数读取视频的某帧，并显示，并写入视频。然后等待1个单位时间，如果期间检测到了键盘输入q，则退出，即关闭窗口。

while(1):
    # get a frame
    ret, frame = cap.read()
    # save a frame
    out.write(frame)
    # show a frame
    cv2.imshow("capture", frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

释放摄像头对象和窗口

调用release()释放摄像头，调用destroyAllWindows()关闭所有图像窗口。

cap.release()
out.release()
cv2.destroyAllWindows()  

源代码

整个程序的源代码如下：

# created by Huang Lu
# 27/08/2016 17:05:45 
# Department of EE, Tsinghua Univ.

import cv2
import numpy as np

cap = cv2.VideoCapture(0)

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

while(1):
    # get a frame
    ret, frame = cap.read()
    # save a frame
    out.write(frame)
    # show a frame
    cv2.imshow("capture", frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
cap.release()
out.release()
cv2.destroyAllWindows() 

也可以参考我的GitHub上的，点击这里。

运行结果

本次输出的视频为：https://raw.githubusercontent.com/hlthu/Python/master/Python_OpenCV/Save_Vedio/output.avi

在命令行进入该源程序所在目录后，运行python main.py后即可显示结果。显示结果如下：

参考

• OpenCV-Python-Toturial-中文版.pdf
• https://github.com/hlthu/Python/tree/master/Python_OpenCV/Save_Vedio/
• https://hlthu.github.io/opencv/2016/06/11/python-opencv-9.html