Python结合FFmpeg批量转换图片为YUV格式教程
本文介绍如何利用Python与FFmpeg工具,将JPG等图片批量转换为YUV格式。通过安装ffmpy3库并配置FFmpeg环境,编写Python脚本自动读取图片、调整尺寸并调用FFmpeg生成指定分辨率的YUV文件,适用于视频编码等需要YUV数据集的场景。
在进行视频编码相关任务时,往往需要构建YUV格式的图片数据集。若手头仅有.jpg格式的资源,可通过以下步骤完成格式转换。该方法同样适用于其他常见图片格式,只需在代码中调整相应的文件扩展名即可。
第一步:安装FFmpeg
FFmpeg官方提供了各版本的下载资源,访问地址为:ffmepg官网
Windows用户可从以下链接获取构建版本:https://ffmpeg.zeranoe.com/builds/
第二步:安装FFmpeg的Python接口ffmpy3
请在命令行环境中执行以下安装命令:
pip install ffmpy3
第三步:将下载并解压得到的Windows版FFmpeg中的ffmpeg.exe文件,复制到执行Python代码的同一目录下。
第四步:参考以下Python代码实现转换逻辑:
# -*- coding: utf-8 -*-
import os
from PIL import Image
from ffmpy3 import FFmpeg
in_jpgDatasetPath = 'I:/VOC/trainval'
out_yuvDatasetPath = 'I:/VVC/loop/yuvdataset'
piclist = os.listdir(in_jpgDatasetPath)
for pic in piclist:
picname = pic.split('.')[0]
picpath = os.path.join(in_jpgDatasetPath,pic)
img = Image.open(picpath)
in_wid,in_hei = img.size
out_wid = in_wid//16*16
out_hei = in_hei//16*16
size = '{}x{}'.format(out_wid,out_hei) #输出文件会缩放成这个大小
outname = out_yuvDatasetPath + '/' + picname + '_'+size+ '.yuv'
ff = FFmpeg(inputs={picpath:None},
outputs={outname:'-s {} -pix_fmt yuv420p'.format(size)})
print(ff.cmd)
ff.run()
以上即为利用Python和FFmpeg批量将其他图片转换为.yuv格式的具体方法,希望能为相关开发者提供参考与支持。
本站声明
本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。













