OpenCV-Python教程:从视频文件、相机获取图像、写视频文件(VedioCapture,VedioWrite)
原文链接: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