基于Python OpenCV优化视频画质：指定H.264码率与无损编码

问题描述 #

我用OpenCV捕获摄像头帧存入列表，再保存为图像序列和视频文件。图像画质良好，但视频压缩严重，码率不足4Mbps，有时甚至低于3Mbps。试过多种fourcc编码、WriteGear、ffmpeg-python都未解决，现请教：
a) 如何为H.264文件指定码率；
b) 如何选择无损压缩编码？

运行环境：MacOS 13.0.1，M1芯片MacBook Pro，Python 3.10

用户提供的代码：

# ::::: import libraries and set variables

import cv2, time, os, shutil, datetime as dt
frames = []
frame_dimensions_known = False

# ::::: define file and folder names

output_folder_name = "output"
frame_folder_name = "frames"
video_file_name = "video.mp4"
frame_file_name = "frame_x.jpg"

# ::::: create output folders and paths

path_script = os.path.abspath(os.path.dirname(__file__))
path_output = os.path.join(path_script,output_folder_name)
if not os.path.exists(path_output):
    os.mkdir(path_output)
    path_session_fold
path_session_folder  = os.path.join(
    path_script,
    output_folder_name,
    dt.datetime.now().strftime("%y%m%d_%H%M%S"))
os.mkdir(path_session_folder)
path_video =  os.path.join(path_session_folder,video_file_name)
path_frames = os.path.join(path_session_folder,frame_folder_name)
os.mkdir(path_frames)


# ::::: open webcam stream

vcap = cv2.VideoCapture(0)
start_time = (time.time() * 1000)
if vcap.isOpened() is False:
    print("[Exiting]: Error accessing webcam stream.")
    exit(0)

# ::::: read frames

while True:
    _grabbed, _frame = vcap.read()
    if _grabbed is False:
        print('[Exiting] No more frames to read')
        exit(0)

    # ::::: get frame dimensions on first read
    if not frame_dimensions_known:
        height, width, channels = _frame.shape
        frame_dimensions_known = True

    # ::::: append frame to list
    frames.append(_frame)

    # ::::: process frames
    pass

    # ::::: display frame
    cv2.imshow("cam", _frame)
    if cv2.waitKey(1) == ord("q"):
        break

# ::::: close webcam stream

vcap.release()
cv2.destroyAllWindows()

# ::::: calculate frame rate

stop_time = (time.time() * 1000)
total_seconds = (stop_time - start_time) / 1000
fps = len(frames) / total_seconds

# ::::: open video writer

fourcc = cv2.VideoWriter_fourcc(*'mp4v')
video_writer = cv2.VideoWriter(
    path_video,
    fourcc,
    fps,
    (width, height)
)

# ::::: write frames to disk as files and video

frame_number = 0
for frame in frames:

    # ::::: write video

    video_writer.write(frame)  # Write out frame to video

    # ::::: construct image file path

    frame_number = frame_number + 1
    frame_number_file = str(frame_number).zfill(4)
    frame_name = frame_file_name.replace('x',frame_number_file)
    frame_path = os.path.join(path_frames,frame_name)

    # ::::: write image

    cv2.imwrite(
        frame_path,
        frame,
        [cv2.IMWRITE_JPEG_QUALITY, 100]
    )

# ::::: clean up

video_writer.release
print (f"Written {frame_number} frames at {fps} FPS.")

解决方案 #

a) 为H.264文件指定码率 #

OpenCV默认的VideoWriter不会显式设置码率，需通过FFmpeg后端（Homebrew安装的OpenCV默认带FFmpeg支持）强制指定：

1. 替换原有的fourcc编码为avc1（标准H.264编码，替代原代码的mp4v）；
2. 通过params参数传入码率值（单位：bps，4Mbps对应4194304 bps）；
3. 强制指定FFmpeg后端确保参数生效。

修改后的视频写入初始化代码：

# 替换原fourcc和video_writer初始化部分
fourcc = cv2.VideoWriter_fourcc(*'avc1')
video_writer = cv2.VideoWriter(
    path_video,
    fourcc,
    fps,
    (width, height),
    apiPreference=cv2.CAP_FFMPEG,
    params=[cv2.VIDEOWRITER_PROP_BITRATE, 4194304]  # 4Mbps码率
)

如果OpenCV无FFmpeg支持，执行brew install opencv重新安装即可。

b) 选择无损压缩编码 #

根据兼容性和场景，推荐以下几种适合M1 Mac的无损编码：

1. FFV1（FFmpeg原生无损编码）

• fourcc编码：FFV1
• 优势：完全无损，文件体积远小于RAW，FFmpeg生态兼容性好
• 适合存档、后期处理场景
• 代码示例：

fourcc = cv2.VideoWriter_fourcc(*'FFV1')
# FFV1更适合MKV容器
video_writer = cv2.VideoWriter(
    path_video.replace('.mp4', '.mkv'),
    fourcc,
    fps,
    (width, height),
    apiPreference=cv2.CAP_FFMPEG
)

2. Apple ProRes 4444 XQ（苹果原生无损编码）

• fourcc编码：ap4h
• 优势：苹果生态内兼容性拉满，适配Final Cut Pro等专业编辑软件，画质完全无损
• 代码示例：

fourcc = cv2.VideoWriter_fourcc(*'ap4h')
# ProRes优先使用MOV容器
video_writer = cv2.VideoWriter(
    path_video.replace('.mp4', '.mov'),
    fourcc,
    fps,
    (width, height),
    apiPreference=cv2.CAP_FFMPEG
)

3. 无压缩RAW视频

• fourcc编码：YUV4（YUV420P无压缩）或RGB（RGB无压缩）
• 劣势：文件体积极大（1080p 30fps每秒约130MB），仅适合特殊专业场景
• 代码示例：

fourcc = cv2.VideoWriter_fourcc(*'YUV4')
# AVI容器支持无压缩编码
video_writer = cv2.VideoWriter(
    path_video.replace('.mp4', '.avi'),
    fourcc,
    fps,
    (width, height)
)

额外修复建议 #

原代码中video_writer.release缺少括号，应改为video_writer.release()，否则可能导致视频文件损坏。

内容的提问来源于stack exchange，提问作者bozolino