YUV到JPG编码使用ffmpeg

时间:2016-02-14 06:05:15

标签: c++ ffmpeg multimedia

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

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

av_new_packet(安培; PKT,大小* 3);`

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

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

3.他们在这里填写数据

frame-> data [0] = buffer;

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;

2 个答案:

答案 0 :(得分:0)

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

答案 1 :(得分:0)

如果查看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个字节。需要额外的代码才能看到这一点。