python保存h264格式视频（linux和windows）

一、 问题分析

1. 当对视频做目标检测、目标跟踪或者其他任务时，有时候需要将输出的一些列图像结果保存为视频，为了方便前端页面展示，还要求视频的编码格式是h264的格式。
2. 常用的方法是利用opencv，但是opencv不能直接保存为h264格式的视频。

为此，这里介绍几种方法，可以顺利保存为h264编码格式的视频。

二、windows下利用opencv的方法

1. 下载openh264的dll动态库文件，跟你的程序放在同一个路径下，这样就可以顺利保存为h264编码格式的视频了。
   下载地址：https://github.com/cisco/openh264/releases

• 下载页面：
  

2. 代码

# coding=utf-8
import cv2
import os

save_name = 'output.mp4'
fourcc = cv2.VideoWriter_fourcc(*'h264')
fps = 25
width = 1920
height = 1080
out = cv2.VideoWriter(save_name, fourcc, fps, (width, height))

img_dir = 'your_img_dir'
files = [os.path.join(img_dir, f) for f in os.listdir(img_dir)]
for file in files:
    img = cv2.imread(file)
    out.write(img)
out.release()

三、Linux下用opencv的方法

需要自己重新编译opencv，比较麻烦，直接放弃。（人生苦短，hhh）

四、Windows和Linux通用的方法（推荐）

利用imageio库，测试使用的imageio版本是2.19.3，注意要先安装ffmpeg。
（可能还需要imageio-ffmpeg的python库，测试使用的版本是0.4.5）

直接上代码：

# coding=utf-8
import os
from tqdm import tqdm
import imageio.v2 as iio

output_file = 'output.mp4'

img_dir = r'your_img_dir'
files = [os.path.join(img_dir, f) for f in os.listdir(img_dir)]


out = iio.get_writer(output_file, format='ffmpeg', mode='I', fps=25, codec='libx264', pixelformat='yuv420p')
for file in tqdm(files):
    frame = iio.imread(file)  # RGB format array
    out.append_data(frame)
out.close()

结束。