原文:TensorRT&Sample&Python[fc_plugin_caffe_mnist]

本文是基于TensorRT . . 基础上,关于其内部的fc plugin caffe mnist例子的分析和介绍。 本例子相较于前面例子的不同在于,其还包含cpp代码,且此时依赖项还挺多。该例子展示如何使用基于cpp写的plugin,用tensorrt python 绑定接口和caffe解析器一起工作的过程。该例子使用cuBLAS和cuDNn实现一个全连接层,然后实现成tensorrt plu ...

2019-03-14 19:25 0 1513 推荐指数:

查看详情

TensorRT&Sample&Python[network_api_pytorch_mnist]

本文是基于TensorRT 5.0.2基础上,关于其内部的network_api_pytorch_mnist例子的分析和介绍。 本例子直接基于pytorch进行训练,然后直接导出权重值为字典,此时并未dump该权重;接着基于tensorrt的network进行手动设计网络结构并填充权重。本文 ...

Thu Mar 14 20:19:00 CST 2019 0 1099
TensorRT&Sample&Python[end_to_end_tensorflow_mnist]

本文是基于TensorRT 5.0.2基础上,关于其内部的end_to_end_tensorflow_mnist例子的分析和介绍。 1 引言 假设当前路径为: 其对应当前例子文件目录树为: 2 基于tensorflow生成模型 其中只有2个文件: model ...

Thu Mar 14 04:07:00 CST 2019 0 999
TensorRT&Sample&Python[uff_custom_plugin]

本文是基于TensorRT 5.0.2基础上,关于其内部的uff_custom_plugin例子的分析和介绍。 本例子展示如何使用cpp基于tensorrt python绑定和UFF解析器进行编写plugin。该例子实现一个clip层(以CUDA kernel实现),然后封装成一个 ...

Sat Mar 16 01:57:00 CST 2019 0 2680
TensorRT&Sample&Python[yolov3_onnx]

本文是基于TensorRT 5.0.2基础上,关于其内部的yolov3_onnx例子的分析和介绍。 本例子展示一个完整的ONNX的pipline,在tensorrt 5.0的ONNX-TensorRT基础上,基于Yolov3-608网络进行inference,包含预处理和后处理 ...

Tue Mar 19 22:59:00 CST 2019 8 8695
TensorRT&Sample&Python[introductory_parser_samples]

本文是基于TensorRT 5.0.2基础上,关于其内部的introductory_parser_samples例子的分析和介绍。 1 引言 假设当前路径为: 其对应当前例子文件目录树为: 该例子展示如何使用TensorRT和包含的对应解析器(UFF,Caffe,ONNX解析器 ...

Thu Mar 14 01:13:00 CST 2019 0 1364
使用caffe提供的python接口训练mnist例子

1 首先肯定是安装caffe,并且编译python接口,如果是在windows上,最好把编译出来的python文件夹的caffe文件夹拷贝到anaconda文件夹下面去,这样就有代码自动提示功能,如下: 本文中使用的ide为anaconda安装中自带的spyder,如图所示,将根目录设置 ...

Sun Oct 30 07:51:00 CST 2016 0 3805
caffe 试运行MNIST

转自:http://www.cnblogs.com/NanShan2016/p/5469942.html 编译完caffe后,在D:\caffe\caffe-master\caffe-master\build\x64\Release 中有caffe.exe 训练篇 1 *.sh文件如何执行 ...

Wed Jul 27 21:09:00 CST 2016 3 2486
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM