利用python和ffmpeg 批量将其他图片转换为.yuv格式的方法

时间:2021-05-22

由于跑编码的需要,所以需要制作一个.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 osfrom PIL import Imagefrom 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格式的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章