ruiyoua / tensorrt_custom_plugin

custom plugin for tensorrt, include pRelu, leakyRelu, Slice...

tensorrt_custom_plugin

custom plugin for tensorrt, include pRelu, leakyRelu, Slice... for tensorrt 5.0.2.6, I think it available for tensorrt other version such as 4.0. but for your best performance, use the newest tensorrt version

extras

1. add the inference sample, use the build guide to build your inference.
2. add the arcface caffe model, if you want to run the sample, download the arcface model
3. base inference include
   • uffmodel -> tensorrt model
   • caffemodel-> tensorrt model
   • onnx -> tensorrt model

depends

1. opencv >= 3.0
2. cuda >= 8.0

build

git clone https://github.com/ruiyoua/tensorrt_custom_plugin

mkdir build

cd build

cmake.. && make

run

./bin/tensorrt-custom

tensorrt_custom_plugin's Issues

Resnet50 caffe to tensorrt

Hi, i used your project to transform resnet50 from caffe to tensort.
The PRelu layers before stage3_unit8_relu1 is built ok，
but the PRelu layer after stage3_unit8_relu1 is built wrong.
The log is "could not parse layer type PReLU".
Have you met this problem?