YUV 到 JPG 编码使用 ffmpeg

YUV to JPG encoding using ffmpeg

本文关键字:ffmpeg 编码 JPG YUV      更新时间:2023-10-16

我正在尝试对YVU文件进行编码并将其另存为jpg文件。 但我不明白以下内容

1.为什么数据包大小是大小*3。

av_new_packet(&pkt,size*3(;'

2.In 担心为什么我们使用尺寸*3/2。

if(fread(buffer , 1, size*3/2, ptrInputFile(<=0('

3.他们如何在这里填写数据

帧>数据[0] = 缓冲区;

帧>数据[1] = 缓冲区 + 大小;

帧>数据[2] = 缓冲区 + siz*5/4;

法典:

AVFormatContext *avFrameContext;
AVOutputFormat *avOutputFormat;
AVStream *avStream;
AVCodecContext *avCodecContext;
AVCodec *avCodec;
AVFrame *frame;
AVPacket pkt;
const char *output = "temp.jpg";
FILE *ptrInputFile;
const char *input = "cuc_view_480x272.yuv";
ptrInputFile = fopen(input ,"rb");
if(!ptrInputFile)
    return -1;
avFrameContext = avformat_alloc_context();
avOutputFormat = av_guess_format("mjpeg", NULL, NULL);
if(!avOutputFormat)
    return -1;
avFrameContext->oformat = avOutputFormat;
if(avio_open(&avFrameContext->pb ,output ,AVIO_FLAG_READ_WRITE)<0)
    return -1;

avStream = avformat_new_stream(avFrameContext,NULL);
if(!avStream)
    return -1;
avCodecContext = avStream->codec;
avCodecContext->codec_id = avOutputFormat->video_codec;
avCodecContext->codec_type = AVMEDIA_TYPE_VIDEO;
avCodecContext->pix_fmt = PIX_FMT_YUVJ420P;
avCodecContext->width = 480;
avCodecContext->height = 272;
avCodecContext->time_base.num = 1;
avCodecContext->time_base.den = 25;
avCodec = avcodec_find_encoder(avCodecContext->codec_id);
if(!avCodec)
    return -1;

if(avcodec_open2(avCodecContext ,avCodec,NULL)<0)
    return -1;
frame = av_frame_alloc();
int size = avpicture_get_size(PIX_FMT_YUVJ420P ,avCodecContext->width, avCodecContext->height);
uint8_t *buffer = (uint8_t*)av_malloc(size*sizeof(uint8_t));
avpicture_fill((AVPicture*)frame, buffer, avCodecContext->pix_fmt ,avCodecContext->width, avCodecContext->height);

//write header
avformat_write_header(avFrameContext, NULL);
int siz = avCodecContext->width*avCodecContext->height;
av_new_packet(&pkt,siz*3);
if(fread(buffer , 1, siz*3/2, ptrInputFile)<=0)
    return -1;
frame->data[0] = buffer;
frame->data[1] = buffer + siz;
frame->data[2] = buffer + siz*5/4;

如果你看一下yuv420p(wiki(的格式,数据在文件中的格式为:

As there are 'siz' length of pixels in the image:
siz length of y value
siz/4 length of u value
siz/4 length of v value

因此,对于问题 2:我们有 siz*3/2 的数据长度要读取。

对于问题 3:y 从缓冲区+0 开始,u 从缓冲区+siz 开始,v 从缓冲区+siz*5/4 开始。

至于问题1:我不确定数据是否转换为RGB。如果转换,则每个像素需要 3 个字节。需要其他代码才能看到这一点。

我对你上面提供的代码知之甚少。但是,如果您尝试对yuv视频进行编码并另存为jpeg,则可以在ffmpeg中直接使用以下命令

ffmpeg -f rawvideo -vcodec rawvideo -s <resolution> -r 25 -pix_fmt yuv420p -i video.yuv -preset ultrafast -qp 0 %d.jpg

用视频的分辨率替换<resolution>,例如1920x1080