ffmpeg nvenc编码
花时间研究了一些ffmpeg的nvenc，本来想我已经有了cuvid，然后又搞出来了nvenc，应该可以做个全套的英伟达的转码了，没想到ffmpeg官网下载的动态库没有cuvid，windows上编译cuvid又老是出错，忧了个伤。
1.nvenc编码
h264_nvenc是很容易调出来的，把编码器ffmpeg源码自带的例子的编码器换成h264_nvenc就行了。可是hevc_nvenc就花了我好多时间，感觉调试技术还是差了好多。
#include "stdafx.h" /*
* Video encoding example
*/
static void video_encode_example(const char *filename)
{
AVCodec *codec;
AVCodecContext *c = NULL;
int i, ret, x, y, got_output;
AVFrame *frame;
AVPacket pkt;
uint8_t endcode[] = { , , , 0xb7 }; av_log_set_level(); //AVBufferRef *device_ref = NULL;
//AVBufferRef *hw_frames_ctx = NULL;
//hw_frames_ctx = (AVBufferRef *)av_mallocz(sizeof(AVBufferRef));
//if (!hw_frames_ctx) {
// ret = AVERROR(ENOMEM);
// return ;
//} //ret = av_hwdevice_ctx_create(&device_ref, AV_HWDEVICE_TYPE_CUDA,"CUDA", NULL, 0);
//if (ret < 0)
// return; //hw_frames_ctx = av_hwframe_ctx_alloc(device_ref);
//if (!hw_frames_ctx) {
// av_log(NULL, AV_LOG_ERROR, "av_hwframe_ctx_alloc failed\n");
// ret = AVERROR(ENOMEM);
// return;
//}
//av_buffer_unref(&device_ref); //c->hw_frames_ctx = av_buffer_ref(hw_frames_ctx);
//if (!hw_frames_ctx) {
// av_log(NULL, AV_LOG_ERROR, "av_buffer_ref failed\n");
// ret = AVERROR(ENOMEM);
// return;
//} printf("Encode video file %s\n", filename); /* find the video encoder */
codec = avcodec_find_encoder_by_name("hevc_nvenc");
//codec = avcodec_find_encoder(AV_CODEC_ID_H265);
//codec = avcodec_find_encoder_by_name("h264_nvenc");
//codec = avcodec_find_encoder(AV_CODEC_ID_H264);
if (!codec) {
fprintf(stderr, "Codec not found\n");
exit();
} c = avcodec_alloc_context3(codec);
if (!c) {
fprintf(stderr, "Could not allocate video codec context\n");
exit();
} /* put sample parameters */
c->bit_rate = ;
/* resolution must be a multiple of two */
c->width = ;
c->height = ;
/* frames per second */
c->time_base.num = ;
c->time_base.den = ;
/* emit one intra frame every ten frames
* check frame pict_type before passing frame
* to encoder, if frame->pict_type is AV_PICTURE_TYPE_I
* then gop_size is ignored and the output of encoder
* will always be I frame irrespective to gop_size
*/
c->gop_size = ;
c->max_b_frames = ;
c->pix_fmt = AV_PIX_FMT_YUV420P;//AV_PIX_FMT_CUDA;
c->max_b_frames = ; AVDictionary *param = ;
//H.264
if (codec->id == AV_CODEC_ID_H264) {
av_dict_set(¶m, "preset", "medium", );
av_dict_set(¶m, "tune", "zerolatency", );
}
//H.265
if (codec->id == AV_CODEC_ID_H265 || codec->id == AV_CODEC_ID_HEVC){
//av_dict_set(¶m, "x265-params", "qp=20", 0);
av_dict_set(¶m, "x265-params", "crf=25", );
av_dict_set(¶m, "preset", "fast", );
av_dict_set(¶m, "tune", "zero-latency", );
} /* open it */
if (avcodec_open2(c, codec, ¶m) < ) {
fprintf(stderr, "Could not open codec\n");
system("pause");
exit();
} FILE *f;
f = fopen(filename, "wb");
if (!f) {
fprintf(stderr, "Could not open %s\n", filename);
exit();
} frame = av_frame_alloc();
if (!frame) {
fprintf(stderr, "Could not allocate video frame\n");
exit();
}
frame->format = c->pix_fmt;
frame->width = c->width;
frame->height = c->height; /* the image can be allocated by any means and av_image_alloc() is
* just the most convenient way if av_malloc() is to be used */
ret = av_image_alloc(frame->data, frame->linesize, c->width, c->height,
c->pix_fmt, );
if (ret < ) {
fprintf(stderr, "Could not allocate raw picture buffer\n");
exit();
} /* encode 1 second of video */
for (i = ; i < ; i++) {
av_init_packet(&pkt);
pkt.data = NULL; // packet data will be allocated by the encoder
pkt.size = ; fflush(stdout);
/* prepare a dummy image */
/* Y */
for (y = ; y < c->height; y++) {
for (x = ; x < c->width; x++) {
frame->data[][y * frame->linesize[] + x] = x + y + i * ;
}
} /* Cb and Cr */
for (y = ; y < c->height / ; y++) {
for (x = ; x < c->width / ; x++) {
frame->data[][y * frame->linesize[] + x] = + y + i * ;
frame->data[][y * frame->linesize[] + x] = + x + i * ;
}
} frame->pts = i; /* encode the image */
ret = avcodec_encode_video2(c, &pkt, frame, &got_output);
if (ret < ) {
fprintf(stderr, "Error encoding frame\n");
exit();
} if (got_output) {
printf("Write frame %3d (size=%5d)\n", i, pkt.size);
fwrite(pkt.data, , pkt.size, f);
av_packet_unref(&pkt);
}
} /* get the delayed frames */
for (got_output = ; got_output; i++) {
fflush(stdout); ret = avcodec_encode_video2(c, &pkt, NULL, &got_output);
if (ret < ) {
fprintf(stderr, "Error encoding frame\n");
exit();
} if (got_output) {
printf("Write frame %3d (size=%5d)\n", i, pkt.size);
fwrite(pkt.data, , pkt.size, f);
av_packet_unref(&pkt);
}
} /* add sequence end code to have a real MPEG file */
fwrite(endcode, , sizeof(endcode), f);
fclose(f); avcodec_close(c);
av_free(c);
av_freep(&frame->data[]);
av_frame_free(&frame);
printf("\n");
} int main(int argc, char **argv)
{
/* register all the codecs */
avcodec_register_all(); avcodec_register_all(); avdevice_register_all(); avfilter_register_all();
av_register_all();
avformat_network_init(); video_encode_example("test.hevc"); system("pause"); return ;
}
代码中av_log_set_level(64);可以帮助输出中间信息，参数可以自行设置，参数为越大，能输出的信息等级越多，我的问题就是通过这个函数知道的，然后到源码中找对应处才最终解决。编码器为hevc_nvenc时max_b_frames必须为0，即代码中的 c->max_b_frames = 0;另外c->pix_fmt = AV_PIX_FMT_YUV420P;//AV_PIX_FMT_CUDA;这行代码需要注意，设置为AV_PIX_FMT_YUV420P意味着数据是从内存读取的，设置为AV_PIX_FMT_CUDA意味着数据在显存中，AV_PIX_FMT_CUDA与cuvid是一起的，只有编出来的ffmpeg支持cuvid时AV_PIX_FMT_CUDA才有效。
2.用nvenc做转码
由于还没有编出支持cuvid的ffmpeg，所以解码这里就先不用cuvid了，用CPU来解码。其实这样有一个好处，就是对格式的要求低，cuvid对格式的输入是有要求的，用这种方法所有用ffmpeg解码后的数据都可以用nvenc来编码，缺点当然是这样比较慢了。
#include "stdafx.h" #include <stdio.h>
#include <io.h> /*
* Video encoding example
*/
static void video_encode_example(const char *filename)
{
AVCodec *codec;
AVCodecContext *c = NULL;
int i, ret, x, y, got_output;
AVPacket pkt;
uint8_t endcode[] = { , , , 0xb7 }; av_log_set_level(); AVFormatContext *pFormatCtx;
int videoindex;
AVCodecContext *pCodecCtx;
AVCodec *pCodec;
AVFrame *pFrame, *pFrameYUV;
uint8_t *out_buffer; char filepath[] = "H:\\nvenc\\灿烂人生1280.rmvb"; av_register_all();
avformat_network_init();
pFormatCtx = avformat_alloc_context(); if (avformat_open_input(&pFormatCtx, filepath, NULL, NULL) != ){
printf("Couldn't open input stream.\n");
return ;
}
if (avformat_find_stream_info(pFormatCtx, NULL)<){
printf("Couldn't find stream information.\n");
return ;
}
videoindex = -;
for (i = ; i<pFormatCtx->nb_streams; i++)
if (pFormatCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO){
videoindex = i;
break;
} if (videoindex == -){
printf("Didn't find a video stream.\n");
return ;
} pCodecCtx = pFormatCtx->streams[videoindex]->codec;
pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
if (pCodec == NULL){
printf("Codec not found.\n");
return ;
}
if (avcodec_open2(pCodecCtx, pCodec, NULL)<){
printf("Could not open codec.\n");
return ;
} pFrame = av_frame_alloc();
pFrameYUV = av_frame_alloc();
out_buffer = (uint8_t *)av_malloc(avpicture_get_size(AV_PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height));
avpicture_fill((AVPicture *)pFrameYUV, out_buffer, AV_PIX_FMT_YUV420P, pCodecCtx->width, pCodecCtx->height); //Output Info-----------------------------
printf("--------------- File Information ----------------\n");
av_dump_format(pFormatCtx, , filepath, );
printf("-------------------------------------------------\n");
struct SwsContext *img_convert_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt,
pCodecCtx->width, pCodecCtx->height, AV_PIX_FMT_YUV420P, SWS_BICUBIC, NULL, NULL, NULL); printf("Encode video file %s\n", filename); /* find the video encoder */
codec = avcodec_find_encoder_by_name("hevc_nvenc");
//codec = avcodec_find_encoder(AV_CODEC_ID_H265);
//codec = avcodec_find_encoder_by_name("h264_nvenc");
//codec = avcodec_find_encoder(AV_CODEC_ID_H264);
if (!codec) {
fprintf(stderr, "Codec not found\n");
exit();
} c = avcodec_alloc_context3(codec);
if (!c) {
fprintf(stderr, "Could not allocate video codec context\n");
exit();
} /* put sample parameters */
c->bit_rate = pCodecCtx->bit_rate;
/* resolution must be a multiple of two */
c->width = pCodecCtx->width;
c->height = pCodecCtx->height;
c->time_base = pCodecCtx->time_base;
c->gop_size = pCodecCtx->gop_size; c->pix_fmt = AV_PIX_FMT_YUV420P;
c->max_b_frames = ; AVDictionary *param = ;
//H.264
if (codec->id == AV_CODEC_ID_H264) {
av_dict_set(¶m, "preset", "medium", );
av_dict_set(¶m, "tune", "zerolatency", );
}
//H.265
if (codec->id == AV_CODEC_ID_H265 || codec->id == AV_CODEC_ID_HEVC){
//av_dict_set(¶m, "x265-params", "qp=20", 0);
av_dict_set(¶m, "x265-params", "crf=25", );
av_dict_set(¶m, "preset", "fast", );
av_dict_set(¶m, "tune", "zero-latency", );
} /* open it */
if (avcodec_open2(c, codec, ¶m) < ) {
fprintf(stderr, "Could not open codec\n");
system("pause");
exit();
} FILE *f;
f = fopen(filename, "wb");
if (!f) {
fprintf(stderr, "Could not open %s\n", filename);
exit();
} AVPacket *packet;
packet = (AVPacket *)av_malloc(sizeof(AVPacket));
int got_picture;
int iCount = ;
int64_t iStart = av_gettime();
while (av_read_frame(pFormatCtx, packet) >= ){
if (packet->stream_index == videoindex){
ret = avcodec_decode_video2(pCodecCtx, pFrame, &got_picture, packet);
if (ret < ){
printf("Decode Error.\n");
return ;
}
if (got_picture){
sws_scale(img_convert_ctx, (const uint8_t* const*)pFrame->data, pFrame->linesize, , pCodecCtx->height,
pFrameYUV->data, pFrameYUV->linesize); av_init_packet(&pkt);
pkt.data = NULL; // packet data will be allocated by the encoder
pkt.size = ; pFrameYUV->width = c->width;
pFrameYUV->height = c->height;
pFrameYUV->format = c->pix_fmt; /* encode the image */
ret = avcodec_encode_video2(c, &pkt, pFrameYUV, &got_output);
if (ret < ) {
fprintf(stderr, "Error encoding frame\n");
exit();
} if (got_output) { iCount++;
fwrite(pkt.data, , pkt.size, f);
av_packet_unref(&pkt); if (iCount % == )
{
printf("1000帧用时：%d 平均每秒 %f 帧 \n", (av_gettime() - iStart)/, (double) * / (av_gettime() - iStart));
printf("Write frame %3d (size=%5d)\n", i, pkt.size); int fd = _fileno(f); //获取文件描述符
_commit(fd); iStart = av_gettime();
} } }
}
av_free_packet(packet);
} /* get the delayed frames */
for (got_output = ; got_output; i++) {
fflush(stdout); ret = avcodec_encode_video2(c, &pkt, NULL, &got_output);
if (ret < ) {
fprintf(stderr, "Error encoding frame\n");
exit();
} if (got_output) {
printf("Write frame %3d (size=%5d)\n", i, pkt.size);
fwrite(pkt.data, , pkt.size, f);
av_packet_unref(&pkt);
}
} /* add sequence end code to have a real MPEG file */
fwrite(endcode, , sizeof(endcode), f);
fclose(f); avcodec_close(c);
av_free(c);
printf("\n");
} int main(int argc, char **argv)
{
/* register all the codecs */
avcodec_register_all(); avcodec_register_all(); avdevice_register_all(); avfilter_register_all();
av_register_all();
avformat_network_init(); video_encode_example("H:\\nvenc\\test.hevc"); system("pause"); return ;
}
上面用nvenc编出来的并没有封装成文件。h264格式potplayer还可以直接播放，hevc格式就不行了。hevc必须先封装，比如封装成mp4文件，然后才能播放。如何封装的例子网上比较多，我这里给个ffmpeg命令行的示例：
ffmpeg -i f:\25国. -c:v copy -f mp4 f:\25国.mp4
最后，nvenc对显卡的要求好像比较高，注意查看自己的显卡是否支持nvenc。
工程源码：http://download.csdn.net/download/qq_33892166/9840113
源码的ffmpg是64位的。
ffmpeg nvenc编码的更多相关文章
- 【视频开发】【CUDA开发】ffmpeg nvenc编码
花时间研究了一些ffmpeg的nvenc,本来想我已经有了cuvid,然后又搞出来了nvenc,应该可以做个全套的英伟达的转码了,没想到ffmpeg官网下载的动态库没有cuvid,windows上编译 ...
- ffmpeg音频编码
在弄音频采集时,需要设置缓存的大小,如果只是简单的采集和直接播放PCM数据,缓存的大小一般不影响播放和保存. 但是,如果需要使用FFMpeg音频编码,这时,音频缓存的大小必须设置av_samples_ ...
- ffmpeg h264编码 extradata 为空
ffmpeg h264编码的例子前面的文章已经介绍,本来主要讲述影响AVCodecContext extradata是否为 空的配置项.如果要求open编码器以后AVCodecContext extr ...
- Ffmpeg AAC 编码错误 Input contains (near) NaN/+-Inf
Ffmpeg AAC编码 如果传入参Frame的Sample Format 为 AV_SAMPLE_FMT_S16,会出现 错误提示 Input contains (near) NaN/+-Inf,需 ...
- ffmpeg,X264编码结果I帧QP比P帧还大
enc_ctx->profile =FF_PROFILE_H264_MAIN ; enc_ctx->time_base.den = 24; enc_ctx->time_base.nu ...
- 使用ffmpeg视频编码过程中踩的一个坑
今天说说使用ffmpeg在写视频编码程序中踩的一个坑,这个坑让我花了好多时间,回头想想,非常多时候一旦思维定势真的挺难突破的.以下是不对的编码结果: ...
- 基于ffmpeg不同编码方式转码后的psnr对比
一.测试说明: 源文件:1080psrc.mp4 时长:900秒 源文件信息:Video: h264 (High) (avc1 / 0x31637661), yuv420p, 1920x1080 [S ...
- JavaCV FFmpeg H264编码
上次成功通过FFmpeg采集摄像头的YUV数据,这次针对上一次的程序进行了改造,使用H264编码采集后的数据. (传送门) JavaCV FFmpeg采集摄像头YUV数据 采集摄像头数据是一个解码过程 ...
- JavaCV FFmpeg AAC编码
上次成功通过FFmpeg采集麦克风的PCM数据,这次针对上一次的程序进行了改造,使用AAC编码采集后的数据. (传送门) JavaCV FFmpeg采集麦克风PCM音频数据 采集麦克风数据是一个解码过 ...
随机推荐
- 模块讲解----sys
sys:跟python解释器相关的信息 #命令行参数list,第一个元素时程序本身路径 print(sys.argv) 注意:执行脚本时,可以传参数. #退出程序,正常退出时exit(0) sys.e ...
- 详解Java中的clone方法 -- 原型模式
转自: http://blog.csdn.net/zhangjg_blog/article/details/18369201 Java中对象的创建 clone顾名思义就是复制, 在Java语言中, ...
- String类型 在底层剖析，并比较 与StringBuilding 的区别
1.string和 stringbuilder的区别: String在任何语言中,都有它的特殊性,在.NET中也是如此.它属于基本数据类型,也是基本数据类型中唯一的引用类型.字符串可以声明为常量,但是 ...
- 微信小程序 drawImage 问题
好久没写了,其实可写的还是挺多,主要还是懒吧... 最近公司项目使用小程序做序列帧动画,大概有 116 张图,共9.6M. 比较闲的日子里实验了一番,主要有以下几种方法, 1. css backgro ...
- [日志] spring boot + logback 日志输出配置
一.输出日志到控制台 springboot有很多starter(参考starter)方便快速构建项目依赖组件,logback是springboot默认的日志组件,启动即可看到打印在控制台输出的info ...
- 在python3下使用OpenCV 抓取摄像头图像并实时显示3色直方图
以下代码为在Python3环境下利用OpenCV 抓取摄像头的实时图像, 通过OpenCV的 calHist函数计算直方图, 并显示在3个不同窗口中. import cv2 import numpy ...
- MVC 4中的前端渲染 @Helper指令
如果我们需要在一个页面或多个页面显示如人民币格式(后台传回来的无￥)￥的格式化.或是对后台数据作如保留小数个数等处理,这些东西经常要用到,特别是一些NULL值的处理,有可能会出错.这时我们可以通过创建 ...
- 为什么 PHP 程序员应该学习使用 Swoole
最近两个月一直在研究 Swoole,研究成果即将在6.21正式开源发布,这段时间没有来水文章,趁着今天放假来水水吧. 借助这篇文章,我希望能够把 Swoole 安利给更多人.虽然 Swoole 可能目 ...
- 输入n，求1~n累加
最开始可能会使用for循环来计算,现在直接使用等差数据计算和公式:s=(a0+n)*n/2 long sum(int n) { long ret=0: ret = (1+n)* n /2: retur ...
- CentOS 7配置静态IP地址
[root@centos1 ~]# ifconfig -bash: ifconfig: command not found 首先,习惯性的输入echo $PATH(查看当前PATH环境变量,跟DOS的 ...