说在前面的话:
以后就算是正式入住简书了,不知道为啥,就是很随性。写点TensorRT 插件的使用,刚接触不懂得很多,简单的问题自己不知道怎么描述,别人也很忙,只能自己琢磨。希望这篇对你有用,嘿,瞅啥呢,说的就是你。
主题内容:
使用TRT 的插件的方法(现在这种接口好像要淘汰,vision = 5.1.5)加速MTCNN,主要参考Github 现有的 TensorRT 加速的 MTCNN 不是基于插件的,而是使用 scale 和 ReLU 、eltwise-sum 。GitHub的PRelu插件代码。TensorRt - caffe中支持prelu。
在“NvInferPlugin.h”这个文件里面是有插件的直接实现的,而且官方文档介绍的也很清楚,但是代码还没有调好,以后会补上。

“NvInferPlugin.h”
IPluginV2:https://docs.nvidia.com/deeplearning/sdk/tensorrt-developer-guide/index.html#add_custom_layer

官方文档介绍
下面开始正文:
上面参考的几个链接已经讲的差不多了,多余的就不多介绍,先上工程代码,配置好CUDA。OpenCV,TRT,common(TRT,samples文件下)环境应该可以直接运行。