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

Posted 2023-03-29 lokvke

目录

• 一、 问题分析
• 二、windows下利用opencv的方法
• 三、Linux下用opencv的方法
• 四、Windows和Linux通用的方法（推荐）

一、 问题分析

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()

结束。

[opencv][原创]关于opencv-python的cv2保存视频不支持H264格式问题探讨

项目有个不合理要求，能够在chrome浏览器打开播放，但是cv2根本不支持H264，由于版权原因，官方不支持h264格式所以当你使用诸如XVID,MJPG等虽然不影响使用和正常播放，但是就是无法在浏览器里面直接打开观看。查遍全网资料，发现Can you support "H264" codec? · Issue #299 · opencv/opencv-python · GitHub

这个全网精华，但是里面尝试了下都不行，因此我得出结论要解决这个问题，只有2条路可行。

第一条：源码编译

这种方法耗时费力，中途错误百出，走的十分艰难，不推荐

第二条：保存后用ffmpeg在格式转换一下

这种方法简单，但是效率上可能就不行了，大家可以尝试ffmpy3这个库或者直接用subprocess调用转换。

目前其他方法暂时测试不行，我的测试环境都是在

ubuntu18.04下面，windows上暂时没有测试过。如果有其他好办法可以分享在评论中，大家一起探讨一下。