OpenCV-Python教程：从视频文件、相机获取图像、写视频文件(VedioCapture,VedioWrite)

返回Opencv-Python教程

原文链接：http://www.juzicode.com/archives/5380

OpenCV-Python教程：读取图像、显示、写入图像 一文介绍了怎么处理静态图像文件，这篇文件介绍怎么从相机、视频文件、动态图片文件获取图像，以及写入视频文件的方法。

1、从视频文件获取图像

使用cap = cv2.VideoCapture(‘文件名称’)构建视频文件的cap实例。

cap.read()方法逐帧提取视频，每一帧为一幅图像，cap.read()方法返回的是一个二元组，下标0的元素值为True或False，如果为Flase表示读取文件完成。下标1的元素为图像对象，也是一个numpy数组类型的数据。

cap.isOpened()用来检查cap实例是否已打开。

cap.release()释放实例。

下面这个例子中读取视频文件并显示，达到播放的效果，同时加入了waitKey()返回值的判断，输入按键’Q’或者’q’就可以退出循环。在这里waityKey()返回的数值和0xff相与后再和字符的ord()值比较，是为了规避某些系统中waitKey()返回的数值在高字节为非0值的情况。

import cv2
print('VX公众号: 桔子code / juzicode.com')
print('cv2.__version__:',cv2.__version__)

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

while cap.isOpened():
    ret, img = cap.read()
    if ret is not True:
        print("读取完成，退出")
        break      
    #处理img
    cv2.imshow('vedio', img)
    
    #检查按键
    key = cv2.waitKey(20) & 0xff
    if  key == ord('q') or key == ord('Q') :
        break
        
print('cap.isOpened():',cap.isOpened())
cap.release()
print('cap.isOpened():',cap.isOpened())

运行结果：

VX公众号: 桔子code / juzicode.com
cv2.version: 4.5.2
cap.isOpened(): True
cap.isOpened(): False

从运行结果看，cap.release()前cap.isOpened()的返回值一直为True。

2、从动态图片获取图像

在 来看看怎么用OpenCV解构Twitter大牛jagarikin的视觉错觉图 一文中我们看到gif格式图片可以看成视频文件，也可以用VideoCapture(文件名)方法打开gif格式的图片，操作过程和上述普通的视频文件类似。

 

3、从相机获取图像

打开相机需要用相机的设备编号（数值型整数）作为入参传入VideoCapture(相机编号)，比如cap = cv2.VideoCapture(0)构建编号为0的相机访问实例，第2台相机则传入1，以此类推，后续步骤的处理方法和读取视频文件一样。

import cv2
print('VX公众号: 桔子code / juzicode.com')
print('cv2.__version__:',cv2.__version__)

cap = cv2.VideoCapture(0)

while cap.isOpened():
    ret, img = cap.read()
    if ret is not True:
        print("读取失败，退出")
        break      
    #处理img
    cv2.imshow('vedio', img)
    
    #检查按键
    key = cv2.waitKey(20) & 0xff
    if  key == ord('q') or key == ord('Q') :
        break
        
print('cap.isOpened():',cap.isOpened())
cap.release()
print('cap.isOpened():',cap.isOpened())
 

 

4、写入视频文件

写视频文件需要创建VideoWriter对象，依次传入4个入参：

• 第1个为文件名称；
• 第2个为编码方式，其中编码方式和文件名称后缀有对应关系；
• 第3个参数为每秒写入的帧数，参考数值为25，符合人眼习惯；
• 第4个参数是图像大小，int类型；

常用的文件名称后缀和编码方式的对应关系有：

文件后缀	编码方式
avi	XVID
avi	MJPG
avi	mp4v（小写）
mp4	mp4v（小写）

编码方式需要创建VideoWriter_fourcc对象，比如可以用下面的2种方式创建MJPG类型的编码：

fourcc=cv2.VideoWriter_fourcc('M','J','P','G')
fourcc=cv2.VideoWriter_fourcc(*'MJPG')

图像大小可以通过cat.get(propId)方法获取，但是该方法获取的是float类型，需要转换为int类型再传入VideoWriter。

cap = cv2.VideoCapture(0)
width = cap.get(cv2.CAP_PROP_FRAME_WIDTH) 
height = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)
width = int(width)
height = int(height)

下面是一个创建XVID、MJPG、mp4v等3种编码方式视频文件的例子：

import cv2
print('VX公众号: 桔子code / juzicode.com')
print('cv2.__version__:',cv2.__version__)
 
#获取图像宽高
cap = cv2.VideoCapture(0)
width = cap.get(cv2.CAP_PROP_FRAME_WIDTH) 
height = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)
width = int(width)
height = int(height)
print(width,height)
 
#创建VideoWriter对象
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('output.avi', fourcc, 25.0, (width,  height))
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
out2 = cv2.VideoWriter('output2.avi', fourcc, 25.0, (width,  height))
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
out3 = cv2.VideoWriter('output3.mp4', fourcc, 25.0, (width,  height))

while cap.isOpened():   
    ret, img = cap.read()
    print(img.shape)
    if ret is not True:
        print("读取失败，退出")
        break      
    #处理img
    cv2.imshow('vedio', img)
    out.write(img)
    out2.write(img)
    out3.write(img)

    #检查按键
    key = cv2.waitKey(1) & 0xff
    if  key == ord('q') or key == ord('Q') :
        break
 
cap.release()
out.release()
out2.release() 
out3.release()

 

原文链接：http://www.juzicode.com/archives/5380

 

扩展阅读：

1. OpenCV-Python教程
2. OpenCV-Python教程：读取图像、显示、写入图像