opencv-python将图片序列转成视频

在跟踪结果以图片形式保存后，有时为了演示会希望将图片序列转换成视频，甚至作成动图插入到博客中，所以这里记录一下自己用的脚本，方便后续取用：
我的cv2版本为4.1.0

img2video.py

import os
import cv2

# image path
im_dir = '/home/lsm/PycharmProjects/py-MDNet/results/Biker/figs'
# output video path
save_video_dir = '/home/lsm/PycharmProjects/py-MDNet/demo'
if not os.path.exists(save_video_dir):
    os.makedirs(save_video_dir)
# set saved fps
fps = 20
# get frames list
frames = sorted(os.listdir(im_dir))
# w,h of image
img = cv2.imread(os.path.join(im_dir, frames[0]))
img_size = (img.shape[1], img.shape[0])
# get seq name
seq_name = os.path.dirname(im_dir).split('/')[-1]
# splice video_dir
video_dir = os.path.join(save_video_dir, seq_name + '.avi')
fourcc = cv2.VideoWriter_fourcc('M', 'J', 'P', 'G')
# also can write like:fourcc = cv2.VideoWriter_fourcc(*'MJPG')
# if want to write .mp4 file, use 'MP4V'
videowriter = cv2.VideoWriter(video_dir, fourcc, fps, img_size)

for frame in frames:
    f_path = os.path.join(im_dir, frame)
    image = cv2.imread(f_path)
    videowriter.write(image)
    print(frame + " has been written!")

videowriter.release()

• 首先指定图片序列路径：比如我的/home/lsm/PycharmProjects/py-MDNet/results/Biker/figs，如下图所示：
  
• 然后指定输出路径
• 之后可以设定视频的帧数FPS，和视频的分辨率(默认就是图片的分辨率)，视频的名称可以自己修改(上面就保存为序列的名字Biker)
• 然后读入一帧帧图片进行视频的写入，就可以生成.avi视频文件了：
  
  后续可以用网上的在线转换工具来生成gif文件。

放一下官网的例子：