TensorRT插件的使用(加速MTCNN)

说在前面的话:

        以后就算是正式入住简书了,不知道为啥,就是很随性。写点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文件下)环境应该可以直接运行。

最后编辑于 :
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容