利用python和ffmpeg 批量将其他图片转换为.yuv格式的方法
时间：2021-09-04 09:49:34|栏目：Python代码|点击： 次
由于跑编码的需要，所以需要制作一个.yuv格式的图片数据集，但是手头只有.jpg格式的，故记录下转换过程。其他图片格式也可以，代码里修改一下就行。
①安装ffmpeg
官网（各种版本）：ffmepg官网
window版：https://ffmpeg.zeranoe.com/builds/
②安装ffmpeg的python3接口ffmpy3
在命令行中输入以下命令：
pip install ffmpy3
③将下载好的windows版解压，把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和GDAL给图片加坐标系的实现思路(坐标投影转换)
栏 目：Python代码
本文标题：利用python和ffmpeg 批量将其他图片转换为.yuv格式的方法
本文地址：http://www.codeinn.net/misctech/172528.html
