将jpg等格式的图片转化为YUV序列(OpenCV、FFmpeg、MATLAB)

将jpg等格式的图片转化为YUV序列有多种,简单介绍其中三种(网上也有很多资料)

1、FFmpeg和Python

原文:https://blog.csdn.net/yuejisuo1948/article/details/83512539
由于跑编码的需要,所以需要制作一个.yuv格式的图片数据集,但是手头只有.png/.jpg格式的,故记录下转换过程。其他图片格式也可以,代码里修改一下就行。

单个文件转换的命令为:

ffmpeg -i xxx.png -s WxH -pix_fmt yuv420p10le xxx.yuv

-i表示输入的图片的路径 -s表示输出图片的分辨率

-pix_fmt是图片格式,详见https://blog.csdn.net/yuejisuo1948/article/details/83617359

最后面是输出图片的路径

批量转换可以写bat文件,也可以直接用以下的方式:

①安装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()
2、Opencv和Python

原文:https://blog.csdn.net/lql0716/article/details/53412252

import cv2

fps = 24   #视频帧率
fourcc = cv2.cv.CV_FOURCC('M','J','P','G')  
videoWriter = cv2.VideoWriter('D:/testResults/match/flower2.avi', fourcc, fps, (1360,480))   #(1360,480)为视频大小
for i in range(1,300):
    p1=0
    p2=i
    img12 = cv2.imread('D:/testResults/img_'+str(p1)+'_'+str(p2)+'.jpg')
#    cv2.imshow('img', img12)
#    cv2.waitKey(1000/int(fps))
    videoWriter.write(img12)
videoWriter.release()

3、MATLAB

原文:https://blog.csdn.net/jxlijunhao/article/details/19981237?utm_source=tuicool&utm_medium=referral

function video=frames2Video(framesPath,videoName,quality,Compressed,fps,startFrame,endFrame)
%framesPath :图像序列所在路径,同时要保证图像大小相同
%videoName:  表示将要创建的视频文件的名字
%quality:    生成视频的质量 0-100
%Compressed: 压缩类型, 'Indeo3'(默认), 'Indeo5', 'Cinepak', 'MSVC', 'RLE' or 'None'
%fps: 帧率
%startFrame ,endFrame ;表示从哪一帧开始,哪一帧结束
 
if(exist('videoName','file'))
    delete videoName.avi
end
 
%生成视频的参数设定
aviobj=avifile(videoName);  %创建一个avi视频文件对象,开始时其为空
aviobj.Quality=quality;
aviobj.Fps=fps;
aviobj.compression=Compressed;
 
%读入图片
for i=startFrame:endFrame
    fileName=sprintf('%08d',i);    %根据文件名而定 我这里文件名是00000001.jpg 00000002.jpg ....
    frames=imread([framesPath,fileName,'.jpg']);
    aviobj=addframe(aviobj,uint8(frames));
end
aviobj=close(aviobj); % 关闭创建视频
end

在command window里输入

path='你的文件路径';
frames2Video(path,'dog',90,'None',5,1,32);
你可以使用libyuv库中的函数来完成RGB32到YUV格式换。以下是使用libyuv库进行RGB32到YUV格式换的示例代码: ```c++ #include "libyuv.h" int main() { // 输入RGB32格式图片 uint8_t* src_rgb32 = ...; int src_stride_rgb32 = ...; int src_width = ...; int src_height = ...; // 分配目标YUV格式的内存 uint8_t* dst_y = new uint8_t[src_width * src_height]; uint8_t* dst_u = new uint8_t[src_width * src_height / 4]; uint8_t* dst_v = new uint8_t[src_width * src_height / 4]; int dst_stride_y = src_width; int dst_stride_u = src_width / 2; int dst_stride_v = src_width / 2; // RGB32格式YUV格式 libyuv::ARGBToI420(src_rgb32, src_stride_rgb32, dst_y, dst_stride_y, dst_u, dst_stride_u, dst_v, dst_stride_v, src_width, src_height); // 输出YUV格式图片 uint8_t* dst_yuv = new uint8_t[src_width * src_height * 3 / 2]; int dst_stride_yuv = src_width; libyuv::I420ToNV12(dst_y, dst_stride_y, dst_u, dst_stride_u, dst_v, dst_stride_v, dst_yuv, dst_stride_yuv, src_width, src_height); // 释放内存 delete[] dst_y; delete[] dst_u; delete[] dst_v; delete[] dst_yuv; return 0; } ``` 在这个示例中,我们首先定义了一个RGB32格式的输入图片,然后使用libyuv库中的ARGBToI420函数将其换为YUV格式。ARGBToI420函数的参数分别表示输入图片的地址、步长、输出YUV格式的Y、U、V分量的地址、步长、输入图片的宽度和高度。换完成后,我们使用libyuv库中的I420ToNV12函数将YUV格式换为NV12格式,最后分配了YUV格式和NV12格式的内存,并在程序结束时释放了内存。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值