通过摄像头捕获视频
我们经常会要用摄像头来捕获实时视频流。OpenCV提供了一个十分简单的接口来做这个。让我们捕获一段摄像头里的视频,把它转换到灰度视频然后显示它。
要捕获一段视频,你需要创建一个VideoCapture对象。它的参数可以是设备索引或者是视频文件名。设备索引是指定哪个摄像头。一般来说只有一个摄像头。所以我们只需要穿入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/False)。如果帧读入正确,会返回True。所以你可以通过检查这个返回值来判断视频是否结束。
有时候,cap可能没有初始化捕获。在这种情况下,这个代码会报错。你可以通过方法cap.isOpened()来检查它是否初始化了。如果是True,那就是初始化了。否则通过cap.open()打开。
你也可以使用cap.get(propId)来访问这个视频的一些属性,propId是从0到18 的一个数。每个数字代表了视频的一个属性。有些值通过cap.set(propId, value)可以修改。
比如,我可以通过cap.get(3)和cap.get(4)来获得这一帧的宽度和高度。它默认会返回640x480,但是我想把值修改成320x240.只需要用ret = cap.set(3, 320)和ret = cap.set(4, 240)
注意:
如果你得到错误,确保摄像头是在正常工作的。
从文件播放视频
和从摄像头捕获一样,值需要把摄像头索引号换成视频文件名。当显示帧的时候,给cv2.waitKey()使用合适的时间。如果太少,视频会太快,如果太高,视频会很慢(慢速播放)。正常情况下25毫秒是ok的。
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()
注意:
确认ffmpeg或者gstreamer安装正确,有时候,处理视频是很头疼的事情
保存一个视频
我们捕获的视频,一帧帧的处理,现在我们想保存视频,对于图片,十分简单,值需要cv2.imwrite().但视频需要多一点工作了。
我们这次创建一个VideoWriter对象。我们应该指定输出文件名(比如:output.avi),然后我们应该指定FourCC代码(下段详细介绍)。 还有每秒帧数(fps)和帧的大小。最后是isColor标志。如果是True,编码器需要彩色帧。否则就是灰度帧。
FourCC是一个4为的代码,用来指定视频编码。可用的视频编码列表可以在fourcc.org找到。这个是跨平台的。下面这些编码都能用:
·In Fedora: DIVX, XVID, MJPG, X264, WMV1, WMV2. (XVID 更推荐. MJPG results in high size video. X264 gives very small size video)
In Windows: DIVX (More to be tested and added)
FourCC代码通过 cv2.VideoWriter_fourcc('M', 'J', 'P', 'G')传入,或者cv2.VideoWriter_fourcc(*'MJPG')
下面的代码从摄像头捕获视频,每一帧都垂直翻转,然后保存
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()