TensorRT教程16：从框架导入模型时使用自定义层--示例

从框架导入模型时使用自定义层

示例：使用caffe模型导入使用自定义层

官方实例

https://github.com/NVIDIA/TensorRT/tree/master/samples/opensource/samplePlugin

https://github.com/NVIDIA/TensorRT/tree/master/samples/opensource/sampleFasterRCNN

示例：使用uff模型导入使用自定义层

官方示例

https://github.com/NVIDIA/TensorRT/tree/master/samples/opensource/sampleUffSSD

https://docs.nvidia.com/deeplearning/tensorrt/sample-support-guide/index.html#uff_custom_plugin

https://docs.nvidia.com/deeplearning/tensorrt/sample-support-guide/index.html#uff_ssd

示例：使用onnx模型导入使用自定义层

使用这个工具可以修改onnx层 ONNX GraphSurgeon utility.

https://docs.nvidia.com/deeplearning/tensorrt/sample-support-guide/index.html#onnx_packnet