this wrapper can support PRelu？

Question

this wrapper can support PRelu？

SyGoing opened this issue 6 years ago · comments

lewes6369 · Answer 1 · Sat Jan 26 2019 13:43:16 GMT+0800 (China Standard Time)

HI,@SyGoing. For PRelu, you can just call createPReLUPlugin in PluginFactory.h checked by the layer name. It is implemented by tensorRT, but you need to load the alpha param.

SyGoing · Answer 2 · Sat Jan 26 2019 14:01:51 GMT+0800 (China Standard Time)

Thanks for your reply, I have implement the prelu with the pluginFactory while I didn't use the createPReLUPlugin ,since I don't know how to use it . I will read your code again.

Rencently, I have seen somebody using the relu and scale to implement the PRelu Layer for keeping the TensorRT's CBR .

lewes6369 · Answer 3 · Sat Jan 26 2019 14:10:06 GMT+0800 (China Standard Time)

Thanks for your sharing.In my code I just used as LeakyRelu.
Could you share the url of the work mentioned above ? What is the network tested?

SyGoing · Answer 4 · Sat Jan 26 2019 14:15:27 GMT+0800 (China Standard Time)

My PRelu implementation have not put on the github,while the other one have used the relu and scale method which is good for mtcnn's prelu layer: https://github.com/PKUZHOU/MTCNN_FaceDetection_TensorRT

lewes6369 · Answer 5 · Sat Jan 26 2019 14:30:32 GMT+0800 (China Standard Time)

Thanks. I will have a look :)