Python 视频帧的获取、保存以及图片序列合成视频

一、视频帧的获取

  首先我是用的环境是python 3.7。获取视频帧可以使用imageio库也可以使用opencv-python库，本次主要介绍第二种，第一种方法参考连接。代码如下：

import cv2                     #引入opencv-python库
cap = cv2.VideoCapture(path)   #获取视频
success = True
print cap.isOpened()           #视频是否打开
while(success): 
   success, frame = cap.read() #循环获取视频帧
cap.release()                  #释放视频资源

cv2.VideoCapture(0)代表打开笔记本或电脑默认摄像头，读取摄像头视频。处理程序放在while循环里进行。

二、视频帧保存为图像序列

 cv2.imwrite('path'+str(num).zfill(4)+'.jpg',frame)
 num=num+1

  首先要注意的是要将整数部分位数通过.zfill()设定为最大num的位数，不然在读取图片序列时会出现读完1.jpg后会先读10.jpg，一直到19.jpg读完后再读2.jpg。这跟内部排序规则有关。以四位为例，当设定位数后10.jpg会改保存为0010.jpg；2.jpg会保存为0002.jpg，那么系统就会按照顺序读取图片序列。此外num要初始化。

三、图片序列合成视频

import cv2
import os

im_dir = 'path'  # 图片存储路径
video_dir = 'compound.avi' #合成后的视频名称, 只能合成avi格式视频
imglist = sorted(os.listdir(im_dir)) #将排序后的路径返回到imglist列表中
img = cv2.imread(os.path.join(im_dir,imglist[0])) #合并目录与文件名生成图片文件的路径,随便选一张图片路径来获取图像大小
H, W, D = img.shape #获取视频高\宽\深度
print('height:' + str(H)+'--'+'width:'+str(W)+'--'+'depth:'+str(D))
fps = 24 #帧率一般选择20-30
img_size = (W,H) #图片尺寸宽x高,必须是原图片的size,否则合成失败
fourcc = cv2.VideoWriter_fourcc('M','J','P','G')
videoWriter = cv2.VideoWriter(video_dir, fourcc, fps, img_size)
for image in imglist:
    img_name = os.path.join(im_dir, image)
    frame = cv2.imread(img_name)
    videoWriter.write(frame)
    print('合成==>'+img_name)
videoWriter.release()
print('finish!')

  合成的视频会保存在与程序同一目录下。

Citation

[1] https://blog.csdn.net/zZzZzZ__/article/details/102666075?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-2.control&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-2.control