欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2023-12-10 10:27:40
由于跑编码的需要,所以需要制作一个.yuv格式的图片数据集,但是手头只有.jpg格式的,故记录下转换过程。其他图片格式也可以,代码里修改一下就行。 ①安装ffmpeg...

由于跑编码的需要,所以需要制作一个.yuv格式的图片数据集,但是手头只有.jpg格式的,故记录下转换过程。其他图片格式也可以,代码里修改一下就行。

①安装ffmpeg

官网(各种版本):

window版:

②安装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格式的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。