opencv把一系列图像保存为视频、一系列图像保存为gif

一. 图像转为视频

import cv2
import os

save_name = 'demo.avi'
fourcc = cv2.VideoWriter_fourcc(*'DIVX')
fps = 30
width = 1920
height = 1080
out = cv2.VideoWriter(save_name, fourcc, fps, (width, height))

img_dir = r'D:\data\2DMOT2015\test\ADL-Rundle-1\img1'
for r, ds, fs in os.walk(img_dir):
    for f in fs:
        file_name = os.path.join(r, f)
        img = cv2.imread(file_name)
        out.write(img)
out.release()

可以看到，其实就是读取每一张图像，然后写到cv2.VideoWriter这个对象里，保存完释放。

二. 序列图像保存为gif

利用imageio库保存，假设图像保存在名字为data的文件夹

import os
import cv2
import imageio

path = 'data'
files = [os.path.join(path, f) for f in os.listdir(path)]
frames = []

for f in files:
	img = imageio.imread(f)  # RGB格式的array
	frames.append(img)
imageio.mimsave('res.gif', frames, fps=20)

读取图像文件也可以用cv2，但是要记得转为RGB格式，再保存为gif。

三. 读取摄像头

import cv2
cap = cv2.VideoCapture(0)
while True:
    ret, frame = cap.read()
    cv2.imshow("capture", frame)
    if cv2.waitKey(50) & 0xFF == ord('q'):
        break
cap.release()
cv2.destroyAllWindows()

结束。