读取视频流与保存视频
要点
- opencv常用读视频函数 cv2.VideoCapture 、cv2.VideoCapture.get 等,可以参考这里
- opencv常用写视频函数 cv2.VideoWriter 等可以参考这里 ,其中视频格式可以参考这里
- videoCapture.read() 是按帧读取视频,ret,frame 是获 .read() 方法的两个返回值。其中 ret 是布尔值,如果读取帧是正确的则返回True,如果文件读取到结尾,它的返回值就为False。frame 就是每一帧的图像,是个三维矩阵。
- 可以使用 cv2.putText 来添加文字,具体参数可以参考这里
代码
- 本例程将读取一个视频,并个视频加上流动水印后保存为一个新视频。
import cv2
#读取视频并获取视频帧率、分辨率、总帧数
VideoCapture = cv2.VideoCapture("VideoExample.mp4")
fps = VideoCapture.get(cv2.CAP_PROP_FPS)
size = (int(VideoCapture.get(cv2.CAP_PROP_FRAME_WIDTH)),
int(VideoCapture.get(cv2.CAP_PROP_FRAME_HEIGHT)))
totalFrames = int(VideoCapture.get(7))
#创建新视频
ViideoWrite = cv2.VideoWriter("VideoWriterExample.avi",cv2.VideoWriter_fourcc('I','4','2','0'),
fps,size)
x=10 #水印坐标
y=10 #水印坐标
i=1
step_x=5
step_y=5
success,frame = VideoCapture.read() #读取视频第一帧
print("第"+str(i)+"帧, 共"+str(totalFrames)+"帧")
while success:
cv2.waitKey(1)
# 给图片添加水印
cv2.putText(frame,'hello,opencv',(x,y),cv2.FONT_HERSHEY_SIMPLEX,1,(255,0,0),3)
cv2.imshow("frame",frame)
ViideoWrite.write(frame) #给新视频添加新帧
# 水印坐标变化
if(x>size[0]):step_x=-5
if(x<0):step_x=5
if(y>size[1]):step_y=-5
if(y<0):step_y=5
x += step_x
y += step_y
success,frame = VideoCapture.read()
i += 1
print("第"+str(i)+"帧, 共"+str(totalFrames)+"帧")
print ('Quitted!') #提示程序已停止
cv2.destroyAllWindows() #程序停止前关闭所有窗口