NVIDIA视频编解码

开源了两个项目:英伟达显卡视频编解码、jetson视频编解码。都是基于官方SDK进行的封装,由于官方自带的demo晦涩难懂并且每块都是独立的,我对SDK进行二次封装并形成了一套较为完整的视频编解码流程,调用简单,有完整的测试程序,想学习SDK或者想直接使用的可以看过来。

1、NVIDIA视频编解码

项目基于英伟达 Video_Codec_SDK_11.0.10 API实现了视频硬解码、渲染、软/硬编码、保存为MP4

  • 解码:使用Video_Codec_SDK_11.0.10解码API对视频进行解码,支持H264、H265。
  • 渲染:使用opencv对图像进行渲染,涉及到了基础的cuda开发(不过都是比较简单的,没有什么难度)。
  • 编码:支持软硬编码切换,硬编码使用Video_Codec_SDK_11.0.10 API,软编码使用ffmpeg API,视频编码格式为H264,如需H265可自行修改(NvCodecRender::EncInit()/NvCodecRender.cpp)。基本nvidia的所有显卡都支持视频解码、但不是所有显卡都支持编码、所以这里实现了软硬编码切换功能。
  • MP4:编码后的视频写入到MP4文件中,这里使用了老陈的libmov库,项目地址:https://github.com/ireader/media-server

下面是项目的测试程序:

#include "NvCodecRender.h"
#include <chrono>
#include <thread>
int main(int argc, char **argv)
{
    if (argc < 5) {
        printf("./demo input output gpu_idx use_nvenc(0  -not use 1- use)\n");
        return -1;
    }
    ck(cuInit(0));
    NvCodecRender *test = new NvCodecRender(argv[1], argv[2], atoi(argv[3]), atoi(argv[4]) == 1 ? true : false);
    auto start_time = std::chrono::high_resolution_clock::now();
    test->Render();
    delete test;
    auto end_time = std::chrono::high_resolution_clock::now();
    auto duration = std::chrono::duration_cast<std::chrono::milliseconds>(end_time - start_time);
    std::cout << "耗时: " << duration.count() << " 毫秒" << std::endl;
    return 0;
}

NVIDIA视频编解码项目地址:https://github.com/BreakingY/Nvidia-Video-Codec

2、JETSON视频编解码

NVIDIA针对自家嵌入式设备jetson的视频编解码又单独实现了一套SDK,我针对这个SDK封装了一个视频编解码库,包含解码和编码库,支持Jetpack 5.x

jetson_dec_5.0.2

  • 基于Jetpack 5.0.2 jetson_multimedia_api 的视频解码库,支持解码H264、H265,实现过程参考jetson_multimedia_api/samples/02_video_dec_cuda
  • 测试程序完善,支持h264裸流、mp4测试文件,h264裸流测试需要修改一下test/video_decode_main.cpp,修改if else分支即可、支持压力测试,可测试jetson解码性能
  • 输入:H264/H265 解码器输出:NV12
  • 生成动态库libJetsonDec.so,头文件 + libJetsonDec.so 即可作为第三方库使用

jetson_enc_5.0.2

  • 基于Jetpack 5.0.2 jetson_multimedia_api 的视频编码库,实现过程参考jetson_multimedia_api/samples/01_video_encode
  • 输入:YUV420P,编码器输出:H264。
  • 如需H265可自行修改(JetsonEnc::encode_proc函数/JetsonEnc.cpp),还是比较简单的
  • 生成动态库libJetsonEnc.so,头文件 + libJetsonEnc.so 即可作为第三方库使用

项目地址:https://github.com/BreakingY/jetpack-dec-enc 

 

posted @   BreakingY  阅读(20)  评论(0编辑  收藏  举报  
相关博文:
阅读排行:
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· 没有源码,如何修改代码逻辑?
· NetPad:一个.NET开源、跨平台的C#编辑器
· PowerShell开发游戏 · 打蜜蜂
· 凌晨三点救火实录:Java内存泄漏的七个神坑,你至少踩过三个!
点击右上角即可分享
微信分享提示