Xavier/Deep Learning/TensorRT/Parsing Caffe: Difference between revisions
mNo edit summary |
mNo edit summary |
||
Line 1: | Line 1: | ||
<noinclude> | <noinclude> | ||
{{Xavier/Head}} | {{Xavier/Head|TensorRT,Parsing,Caffe model,Parsing Caffe model}} | ||
</noinclude> | </noinclude> | ||
<!-- If you want a custom title for the page, un-comment and edit this line: | <!-- If you want a custom title for the page, un-comment and edit this line: |
Revision as of 17:45, 17 November 2018
Parsing Caffe model for TensorRT
The process for caffe models are fairly similar to Tensorflow models. The key difference is that you don't need to generate a uff model file. Caffe model file (.caffemodel) can be imported directly from tensorrt.
Loading a caffe model is an actual example provided by NVIDIA with TensorRT naned sample_mnist. For more details on this example please refer to the C++ API section.