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
video_dir = '/home/lsm/PycharmProjects/py-MDNet/demo'
if not os.path.exists(video_dir):
    os.makedirs(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(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,如下图所示:
    opencv-python将图片序列转成视频_第1张图片
  • 然后指定输出路径
  • 之后可以设定视频的帧数FPS,和视频的分辨率(默认就是图片的分辨率),视频的名称可以自己修改(上面就保存为序列的名字Biker)
  • 然后读入一帧帧图片进行视频的写入,就可以生成.avi视频文件了:
    opencv-python将图片序列转成视频_第2张图片
    后续可以用网上的在线转换工具来生成gif文件。

放一下官网的例子:
opencv-python将图片序列转成视频_第3张图片

你可能感兴趣的:(opencv-python,目标跟踪)