OpenCV是如今最流行的计算机视觉库,我在上一篇文章中已经介绍了如何安装。
1 利用OpenCV打开摄像头
- cv2.VideoCapture(0):要使用摄像头,需要使用cv2.VideoCapture(0)创建VideoCapture对象,参数:0指的是摄像头的编号。如果你电脑上有两个摄像头的话,访问第2个摄像头就可以传入1。
- capture.read(): capture.read()函数返回的第1个参数ret(return value缩写)是一个布尔值,表示当前这一帧是否获取正确。
# 打开摄像头并灰度化显示
import cv2
capture = cv2.VideoCapture(0)
while(True):
# 获取一帧
ret, frame = capture.read()
# 将这帧转换为灰度图
# gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
#第一个参数是窗口名称,第二个参数从摄像头读取到的内容
cv2.imshow('frame', frame)
if cv2.waitKey(1) == ord('q'):
break
capture.release()
cv2.destroyAllWindows()
如果需要灰度显示的话, 只需要简单的修改如下几段代码就可以了:
- cv2.cvtColor(): 用来转换颜色,这里将彩色图转成灰度图。
# 打开摄像头并灰度化显示
import cv2
capture = cv2.VideoCapture(0)
while(True):
# 获取一帧
ret, frame = capture.read()
# 将这帧转换为灰度图
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
#第一个参数是窗口名称,第二个参数从摄像头读取到的内容
cv2.imshow('frame', gray)
if cv2.waitKey(1) == ord('q'):
break
capture.release()
cv2.destroyAllWindows()
通过cap.get(propId)可以获取摄像头的一些属性,比如捕获的分辨率,亮度和对比度等。propId是从0~18的数字,代表不同的属性,完整的属性列表可以参考:Property Identifier。也可以使用cap.set(propId,value)来修改属性值。
我们在
capture = cv2.VideoCapture(0)
加上如下的语句:
# 获取捕获的分辨率
# propId可以直接写数字,也可以用OpenCV的符号表示
width, height = capture.get(3), capture.get(4)
print(width, height)
# 以原分辨率的一倍来捕获
capture.set(cv2.CAP_PROP_FRAME_WIDTH, width * 2)
capture.set(cv2.CAP_PROP_FRAME_HEIGHT, height * 2)
2 利用OpenCV播放本地视频(无声音)
跟打开摄像头一样,如果把摄像头的编号换成视频的路径就可以播放本地视频了
capture = cv2.VideoCapture("D:\\you-get_download\\muzi.flv")
cv2.waitKey():它的参数表示暂停时间,所以这个值越大,视频播放速度越慢,反之,播放速度越快,通常设置为25或30。
import cv2
# 播放本地视频
capture = cv2.VideoCapture("D:\\you-get_download\\muzi.flv")
while(capture.isOpened()):
ret, frame = capture.read()
#gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
cv2.imshow("frame", frame) #彩色画面播放
#cv2.imshow("frame", gray) #黑白画面播放
if cv2.waitKey(30) == ord('q'):
break
capture.release()
cv2.destroyAllWindows()
3 利用OpenCV录制视频
之前我们保存图片用的是cv2.imwrite(),要保存视频,我们需要创建一个VideoWriter的对象,需要给它传入四个参数:
- 输出的文件名,如’output.avi’
- 编码方式FourCC码
- 帧率FPS
- 要保存的分辨率大小
FourCC是用来指定视频编码方式的四字节码,所有的编码可参考Video Codecs。如MJPG编码可以这样写:cv2.VideoWriter_fourcc(*’MJPG’)或cv2.VideoWriter_fourcc(‘M’,’J’,’P’,’G’)
#录制视频
import cv2
capture = cv2.VideoCapture(0)
# 定义编码方式并创建VideoWriter对象
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
outfile = cv2.VideoWriter("D:\\you-get_download\\output.avi", fourcc, 25., (640, 480))
while(capture.isOpened()):
ret, frame = capture.read()
if ret:
outfile.write(frame) # 写入文件
cv2.imshow('frame', frame)
if cv2.waitKey(1) == ord('q'):
break
else: break
capture.release()
cv2.destroyAllWindows()
总结
- 使用cv2.VideoCapture()创建视频对象,然后在循环中一帧帧显示图像。参数传入数字时,代表打开摄像头,传入本地视频路径时,表示播放本地视频
- cap.get(propId)获取视频属性,cap.set(propId,value)设置视频属性
- cv2.VideoWriter()创建视频写入对象,用来录制/保存视频