利用python和ffmpeg 批量将其他图片转换为.yuv格式的方法_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和ffmpeg 批量将其他图片转换为.yuv格式的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持服务器之家。

原文链接：https://blog.csdn.net/yuejisuo1948/article/details/83512539

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

相关文章

热门资讯