Difference between revisions of "Xavier/Deep Learning/TensorRT/Parsing Caffe"
< Xavier | Deep Learning | TensorRT
m |
m |
||
Line 9: | Line 9: | ||
==Parsing Caffe model for TensorRT== | ==Parsing Caffe model for TensorRT== | ||
− | The process for | + | The process for Caffe models is 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 | + | Loading a Caffe model is an actual example provided by NVIDIA with TensorRT named sample_mnist. For more details on this example please refer to the [[Xavier/Deep_Learning/TensorRT/Building_Examples#C.2B.2B_API|C++ API]] section. |
<noinclude> | <noinclude> | ||
{{Xavier/Foot|Deep Learning/TensorRT/Parsing Tensorflow|Deep Learning/TensorRT/Building Examples}} | {{Xavier/Foot|Deep Learning/TensorRT/Parsing Tensorflow|Deep Learning/TensorRT/Building Examples}} | ||
</noinclude> | </noinclude> |
Revision as of 14:16, 1 December 2020
Parsing Caffe model for TensorRT
The process for Caffe models is 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 named sample_mnist. For more details on this example please refer to the C++ API section.