小白菜的opencv学习(2)视频的录制和导入
接下来学习视频的导入。
视频是由图片一帧一帧的放在一起的,所以这里猜测以后学习对视频的处理和对图片的处理过程是一样的。
import numpy as np
import cv2
cap = cv2.VideoCapture(0)
while(True):
ret, frame = cap.read()
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
cv2.imshow('frame',gray)
if cv2.waitKey(1) & 0xFF == ord('q'):
breakcap.release()
cv2.destroyAllWindows()
前两句导入库没毛病。
接下来和图像导入相似,用到了cap = cv2.VideoCapture(0),实参‘0’代表是摄像头0号,多个摄像头调用的时候会用到这个参数。
ret, frame = cap.read()中,ret被定义为一个代表是否有返回的布尔值,frame则被定义为每个返回的帧。 如果没有帧,你不会得到错误,你会得到None。
视频的录制是不断用cap.read()来获取帧frame,然后用gray = cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY)来进行第一步的处理(这里是BGR to GRAY灰度化)。OpenCV 将颜色读取为 BGR(蓝绿色红色),但大多数计算机应用程序读取为 RGB(红绿蓝)。
然后再通过循环用cv2.imshow(‘frame’,gray)將处理之后的帧gray以‘frame’的名称输出。
由于用到了循环,所以程序结束的判断方式和图片的方式不一样了,可以自行理解。
该程序是视频的显示并没有储存的语句。
import numpy as np
import cv2
cap = cv2.VideoCapture(0)
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('output.avi',fourcc, 20.0, (640,480))
while(True):
ret, frame = cap.read()
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
out.write(frame)
cv2.imshow('frame',gray)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
cap.release()
out.release()
cv2.destroyAllWindows()
这里多了几句话:
fourcc = cv2.VideoWriter_fourcc(*‘XVID’)我理解为将‘XVID’转成int型数据。
因为cv::VideoWriter::VideoWriter()中有五个参数,其中第二个int fourcc用四个字符标识来表示编码格式。cv提供了一个函数直接来进行转换。20.0是每一帧的大小。
定义好之后,进入while循环,cvtColor()和imshow()将处理好的每一帧显示出来,而write()将没有处理之前的储存到视频中。
我这里尝试用write()将处理之后的图片gray储存,但是输出的视频不管怎么样只有5.7kb,并且无法播放。我认为这里是因为经过灰度化处理后图像和初始定义的不一致,所以write()不认这个类型,而5.7kb可能是每个视频里面及时一帧也没有也会占的一个空间大小。
如果要录制灰度的视频,可以将cv2.VideoWriter(‘output.avi’,fourcc, 20.0, (640,480))后面加上一个bool型的参数is_color,这里默认的是true,可以改成false,这时候下面的write()就可以传入灰度的图片。
这是灰度的视频录制
import numpy as np
import cv2
cap = cv2.VideoCapture(0)
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('output.avi',fourcc, 20.0, (640,480),False)
while(True):
ret, frame = cap.read()
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
out.write(gray)
cv2.imshow('frame',gray)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
cap.release()
out.release()
cv2.destroyAllWindows()
经过测试,该程序有效,并且如果最后一个参数定义为False后,write()不能识别彩色图像。输出的文件同样也是5.7kb。一定程度上验证了上面的猜想。