[]how to use caffe model with TensorRT c++

 

 

 

复制代码
//IHostMemory *gieModelStream {nullptr};
    //const char* prototxt = "./googlenet/test_20181010.prototxt";//argv[1];
    //const char* caffemodel = "./googlenet/lane_area_lx1890_iter_320000_20181010.caffemodel";//argv[2];
    //std::vector<std::string> output_blobnames;
    //output_blobnames.push_back(OUTPUT_BLOB_NAME_1);
    //output_blobnames.push_back(OUTPUT_BLOB_NAME_2);
    //caffeToGIEModel(prototxt, caffemodel, output_blobnames, 1, &plugin_factory, gieModelStream, true);
    //caffeToGIEModel_serialize(prototxt, caffemodel, output_blobnames, 1, &plugin_factory, gieModelStream, true, s);
    plugin_factory.destroyPlugin();
    //std::vector<std::string>().swap(output_blobnames);
    // deserialize the engine
    IRuntime* runtime = createInferRuntime(gLogger);
    //ICudaEngine* engine = runtime->deserializeCudaEngine(gieModelStream->data(), gieModelStream->size(), &plugin_factory);
    ICudaEngine* engine = runtime->deserializeCudaEngine(ss.data(), ss.size(), &plugin_factory);
    //if (gieModelStream)
        //gieModelStream->destroy();
    IExecutionContext *context = engine->createExecutionContext();
复制代码

 

posted on   鹅要长大  阅读(215)  评论(0编辑  收藏  举报

编辑推荐:
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
阅读排行:
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 提示词工程——AI应用必不可少的技术
· .NET周刊【3月第1期 2025-03-02】

导航

< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

统计

点击右上角即可分享
微信分享提示