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类型;

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

文件后缀编码方式
aviXVID
aviMJPG
avimp4v(小写)
mp4mp4v(小写)

编码方式需要创建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教程:读取图像、显示、写入图像

发表评论

您的电子邮箱地址不会被公开。 必填项已用*标注