流程描述:我把每一帧原始YUV图像数据,构造成Y4M(YUV4MPEG2)格式的流,然后使用ffmpeg编码成h264格式进行RTMP推流。
ffmpeg执行avformat_open_input打开文件和执行avformat_find_stream_info探测流都正常,但在执行avformat_write_header报错:Invalid data found when processing input。
编码器参数设置错误
// enc_ctx编码器上下文,dec_ctx解码器上下文
AVCodecContext *dec_ctx, *enc_ctx;
...
dec_ctx = pFormatCtx->streams[i]->codec;
...
if (dec_ctx->codec_type == AVMEDIA_TYPE_VIDEO) {
LOGD_DEBUG("height = %d, width = %d, sample_aspect_ratio = %d/%d, pix_fmt = %d, framerate = %d/%d, time_base = %d/%d",
dec_ctx->height, dec_ctx->width,
dec_ctx->sample_aspect_ratio.num,
dec_ctx->sample_aspect_ratio.den,
dec_ctx->pix_fmt,
dec_ctx->framerate.num, dec_ctx->framerate.den,
dec_ctx->time_base.num, dec_ctx->time_base.den);
enc_ctx->height = dec_ctx->height;
enc_ctx->width = dec_ctx->width;
enc_ctx->sample_aspect_ratio = dec_ctx->sample_aspect_ratio;
/* take first format from list of supported formats */
if (encoder->pix_fmts)
enc_ctx->pix_fmt = encoder->pix_fmts[0];
else
enc_ctx->pix_fmt = dec_ctx->pix_fmt;
/* video time_base can be set to whatever is handy and supported by encoder */
enc_ctx->time_base = dec_ctx->framerate;
enc_ctx->codec_type = AVMEDIA_TYPE_VIDEO;
}
我这里的编码器参数是根据解码器里参数进行设置的,但解码器取得的部分参数时不准确的,推测原因可能和我构造的Y4M(YUV4MPEG2)流格式有关。
将上面的代码修改为:
// enc_ctx编码器上下文,dec_ctx解码器上下文
AVCodecContext *dec_ctx, *enc_ctx;
...
dec_ctx = pFormatCtx->streams[i]->codec;
...
if (dec_ctx->codec_type == AVMEDIA_TYPE_VIDEO) {
LOGD_DEBUG("height = %d, width = %d, sample_aspect_ratio = %d/%d, pix_fmt = %d, framerate = %d/%d, time_base = %d/%d",
dec_ctx->height, dec_ctx->width,
dec_ctx->sample_aspect_ratio.num,
dec_ctx->sample_aspect_ratio.den,
dec_ctx->pix_fmt,
dec_ctx->framerate.num, dec_ctx->framerate.den,
dec_ctx->time_base.num, dec_ctx->time_base.den);
enc_ctx->height = dec_ctx->height;
enc_ctx->width = dec_ctx->width;
enc_ctx->sample_aspect_ratio = {476, 477};//dec_ctx->sample_aspect_ratio;
/* take first format from list of supported formats */
if (encoder->pix_fmts)
enc_ctx->pix_fmt = encoder->pix_fmts[0];
else
enc_ctx->pix_fmt = dec_ctx->pix_fmt;
/* video time_base can be set to whatever is handy and supported by encoder */
enc_ctx->time_base = av_inv_q({25, 1});//dec_ctx->framerate);
enc_ctx->codec_type = AVMEDIA_TYPE_VIDEO;
}