YUV转JPG编码使用ffmpeg

Question

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

1.why数据包大小为size*3。

av_new_packet(&pkt,size*3);`

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

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

3.how他们正在这里填写数据

帧->数据[0] = 缓冲区；

frame->data[1] = buffer + siz;

frame->data[2] = buffer + 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;

Answer 1

我对你上面提供的代码了解不多。但是如果你想编码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

Answer 2

如果您查看 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从buffer+0开始，u从buffer+siz开始，v从buffer+siz*5/4开始。

关于问题1：我不确定数据是否转换为RGB。如果它被转换，那么每个像素需要 3 个字节。需要额外的代码才能看到。

YUV转JPG编码使用ffmpeg

YUV to JPG encoding using ffmpeg

c++

multimedia

ffmpeg