内容来自OpenCV-Python Tutorials 自己翻译整理 目标: - 读取视频、显示视频、保存视频
- 使用摄像头拍摄并显示
- 学习cv2.VideoCapture(), cv2.VideoWriter()函数
使用摄像头拍摄视频:
(我的用的是笔记本电脑上面的摄像头)
OpenCV 提供了一个非常简单的接口。我们把视频转换成黑白格式,然后显示出来。 首先要创建一个VideoCapture对象。参数可以是设备标示(device index)或是一个视频文件的名称。设备标示是指定使用哪个摄像头的编号。通常一个摄像头会连接上。所以只需要传递参数0或者-1。可以选择第二个摄像头通过传递参数1,以此类推。之后就能通过摄像头获得每一帧的数据,在最后释放摄像头资源。 import numpy as np 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(1) & 0xFF == ord('q'): break
# When everything done, release the capture cap.release() cv2.destroyAllWindows() cap.read() 函数返回一个布尔类型。如果读的视频帧是正确的,返回True,所以可以通过返回值检查。
有时,cap可能没有初始化摄像头,可以检查是否初始化或者使用cap.isOpened() 函数,如果返回True则继续,否则使用cap.open()
可以使用cap.get(proid)函数来设置视频的属性,proid是0到18的数字每个数组表示一个视频的属性,具体见Property Identifier 其中一些属性可以是用cap.set(proid,value)方法修改
例如,使用cap.get(3)和cap.get(4)获得每一帧的宽度和高度,获得的结果是640和480,可以把它修改成320×240 ret = cap.set(3,320) ret = cap.set(4,240)
#!/usr/bin/env python # coding=utf-8
import cv2 import numpy as np from matplotlib import pyplot as plt
cap = cv2.VideoCapture(0) ret = cap.set(3,320) ret = cap.set(4,240) while(True): ret,frame = cap.read() gray = cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY)
cv2.imshow('frame',gray) if cv2.waitKey(1) & 0xFF == ord('q'): break
cap.release()
cv2.destroyAllWindows() (不要把set写到循环里,否则会很卡顿)
播放视频文件: 只需要把VideoCapture里面的摄像头标示改成文件名即可,如果视频很小,播放的太快,可以使用参数cv2.waitKey()函数,设置成25毫秒效果不错
import numpy as np import cv2
cap = cv2.VideoCapture('vtest.avi')
while(cap.isOpened()): ret, frame = cap.read()
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
cv2.imshow('frame',gray) if cv2.waitKey(1) & 0xFF == ord('q'): break
cap.release()
cv2.destroyAllWindows() 存储视频: 要把摄像机拍摄到的视频按帧存起来可以是用cv2.imwrite()函数
首先建立一个VideoWriter对象,然后指定要保存的文件名,例如output.avi。接下来使用FourCC,接下来传递每秒的帧数和帧的大小参数,最后是isColor标记,如果该标记为真,将得到彩色的图片。
FourCC 是用于指定视频编解码器 4 字节代码。代码可以在fourcc.org上找到,该代码是平台相关的。(本人机器为ubuntu 14.04)
import numpy as np import cv2
cap = cv2.VideoCapture(0)
# Define the codec and create VideoWriter object fourcc = cv2.VideoWriter_fourcc(*'XVID') out = cv2.VideoWriter('output.avi',fourcc, 20.0, (640,480))
while(cap.isOpened()): ret, frame = cap.read() if ret==True: frame = cv2.flip(frame,0)
# write the flipped frame out.write(frame)
cv2.imshow('frame',frame) if cv2.waitKey(1) & 0xFF == ord('q'): break else: break
# Release everything if job is finished cap.release() out.release()
cv2.destroyAllWindows() 得到的视频是掉转过来的
|