ffmpeg将多个yuv文件编码为MP4视频文件

news2024/10/6 12:25:05

一、编码方案

在视频录制时，每一帧保存为一个yuv文件，便于纠错和修改。在编码为MP4文件时，我的方案是将所有yuv文件先转码为单个MP4文件，然后使用ffmpeg的concat功能拼接为完整的视频。

二、shell脚本

#!/bin/bash

# 检查参数数量是否正确
if [ "$#" -ne 5 ]; then
    echo "Usage: $0 <directory> <width> <height> <FPS> <quality>"
    echo "quality : ultrafast superfast veryfast faster fast medium slow slower veryslow placebo"
    exit 1
fi

# 将传入的参数赋值给变量
# 使用双引号将变量括起来，防止变量被shell解释
DIRECTORY="$1"
WIDTH="$2"
HEIGHT="$3"
FRAMERATE="$4"
QUALITY="$5"

# 输入输出格式
PIX_FMT="yuv420p"

# 输入文件名前缀
PREFIX="video_frame_"

# 输出文件名    
OUTPUT_VIDEO="output.mp4"

# 进入目录
cd $DIRECTORY || exit 1

# 清理文件列表
rm -f filelist.txt

# 删除临时文件
rm -f temp_*.mp4

# 使用for循环遍历目录中的YUV文件  
for file in ./${PREFIX}*.yuv; do  
    # 提取文件名（不带扩展名）  
    base=$(basename "$file" .yuv)

    # 设置输出文件名
    output_temp="temp_${base}.mp4"

    # 不显示输出
    ffmpeg -s ${WIDTH}x${HEIGHT} -f rawvideo -pix_fmt $PIX_FMT -r $FRAMERATE -i "$file" -c:v libx264 -pix_fmt $PIX_FMT -frames:v 1 "$output_temp" > /dev/null 2>&1
    
    # 显示输出
    #ffmpeg -s ${WIDTH}x${HEIGHT} -f rawvideo -pix_fmt $PIX_FMT -r $FRAMERATE -i "$file" -c:v libx264 -pix_fmt $PIX_FMT -frames:v 1 "$output_temp"
    
    echo "file '$output_temp'" >> filelist.txt
done


# 使用FFmpeg合并所有MP4文件
ffmpeg -f concat -safe 0 -i filelist.txt -c copy "$OUTPUT_VIDEO"

# 检查ffmpeg命令是否执行成功
if [ $? -eq 0 ]; then
    echo "Success! Conversion completed: $OUTPUT_VIDEO"
else
    echo "Error! Conversion failed."
fi

# 删除临时文件
rm -f temp_*.mp4

# 清理文件列表
rm -f filelist.txt

使用filelist.txt保存中间文件名，注意写入格式，前面有一个file，这是concat功能的读取格式。

每一项参数都可以查到，注意要加上 -frames:v 1，不加会报错