TensorRt - caffe中支持prelu

好久没写博客了，今天就更新一下。

我们都知道TensorRT的激活层，支持relu，却不支持prelu。但是总有需要使用prelu层的时候。

补充： prelu层相当于是带参数的relu

运行环境： ubuntu16.04+ TenosrRt 4.0.1以上 + cuda 8.0/cuda9.0 + Cudnn 7.1 + opencv3

一、下载

先去github下载tensorrt-prelu的相关文件

二、将Gplugin.cu重命名为GpluginGPU.cu

否则编译会出错

三、在GpluginGPU.h中 217行中DataType改为nvinfer1：：DataType

即

改为

四、添加prelu

在caffeToGIEModel函数结束的时候，要记得销毁parser

parser->destory();

调用caffeToGIEModel：

五、修改deploy文件

在deploy.protox中

将prelu层的type从prelu改为relu，prelu_param保留

这样就可以在tensorrt中使用prelu层了

TensorRt - caffe中支持prelu相关推荐

怎么在Caffe中配置每一个层的结构
如何在Caffe中配置每一个层的结构最近刚在电脑上装好Caffe,由于神经网络中有不同的层结构,不同类型的层又有不同的参数,所有就根据Caffe官网的说明文档做了一个简单的总结. 1. Vision ...
在caffe中使用hdf5的数据
caffe默认使用的数据格式为lmdb文件格式,它提供了把图片转为lmdb文件格式的小程序,但是呢,我的数据为一维的数据,我也要分类啊,那我怎么办?肯定有办法可以转为lmdb文件格式的,我也看了一些源 ...
阿里巴巴宣布架构调整；英伟达放大招！重磅发布 TensorRT 7 ，支持超千种计算变换；苹果、谷歌和亚马逊罕见结盟……...
戳蓝字"CSDN云计算"关注我们哦! 嗨,大家好,重磅君带来的[云重磅]特别栏目,如期而至,每周五第一时间为大家带来重磅新闻.把握技术风向标,了解行业应用与实践,就交给我重磅君吧 ...
caffe中loss函数代码分析--caffe学习（16）
接上篇:caffe中样本的label一定要从序号0开始标注吗?–caffe学习(15) A: 1:数学上来说,损失函数loss值和label从0开始还是从1或者100开始是没有直接联系的,以欧式距离损 ...
caffe中的layer
layer是神经网络搭建的脚手架,理解了layer,才能盖好神经网络这座摩天大楼. 下图是一张关于layer的思维导图,在功力到达一定程度的时候才可练此功,到时一定会有不一样的收获. 1. Outli ...
Caffe中BN层与CONV层的融合（merge_bn）
半年前写的博客,今天发现没有发出去,还好本地有md的文档,决定重新发一下毕竟网上来回抄袭的blog太多了,代码质量也莫得保证今天需要用pytorch融合下bn层,写个脚本稍后再传到csdn上原理 ...
Caffe中对MNIST执行train操作执行流程解析
之前在 http://blog.csdn.net/fengbingchun/article/details/49849225 中简单介绍过使用Caffe train MNIST的文章,当时只是仿照ca ...
Caffe 中的卷积--权宜之计
Caffe 中的卷积中的效果不是很好,经常被人所诟病.首先来看看作者本人是怎么回答这个问题的. https://github.com/Yangqing/caffe/wiki/Convolution-i ...
SpreadJS 在 Angular2 中支持绑定哪些属性？
SpreadJS 纯前端表格控件是基于 HTML5 的 JavaScript 电子表格和网格功能控件,提供了完备的公式引擎.排序.过滤.输入控件.数据可视化.Excel 导入/导出等功能,适用于 .N ...
[转]使用 .NET Framework 2.0 在您的应用程序中支持证书
NET 安全使用 .NET Framework 2.0 在您的应用程序中支持证书 Dominick Baier 本文讨论: Windows 证书存储区 .NET 中的证书类验证.SSL.Web 服 ...