[]how to use caffe model with TensorRT c++
2024-09-05 06:06:39
//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();
最新文章
- 在Nodejs中如何调用C#的代码
- mongodb版本特性
- 滚动条滚动事件 js
- NSSet、NSMutableSet基本用法
- bzoj2506
- kindeditor-4.1.10在线编辑器的使用[多个]
- python 面向对象简单理解
- php 5.5 xhprof for windows
- 可信执行环境(TEE)介绍
- Struts 框架 之 Hello World
- Java第一章----Java概述+环境搭建
- 初探linux子系统集之timer子系统(三)
- 02_JNI中Java代码调用C代码,Android中使用log库打印日志,javah命令的使用,Android.mk文件的编写,交叉编译
- 基于用户协同过滤--UserCF
- Sqoop export(Hive to MySQL) 的一些 reference
- 用python脚本获取运行环境中的module 列表
- Linux故障:linux中使用ifconfig命令查看网卡信息时显示为eth1,但是在network-scripts中只有ifcfg-eth0的配置文件,并且里面的NAME=";eth0";。
- 在operator =中要处理“自我赋值”
- Intellij IDEA注册激活破解
- spark streaming 整合 kafka(一)