从框架导入模型时使用自定义层
示例:使用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