NVIDIA Xavier - Deep Learning - TensorRT - Parsing Caffe

From RidgeRun Developer Connection
< Xavier‎ | Deep Learning‎ | TensorRT
Revision as of 12:45, 17 November 2018 by Spalli (talk | contribs)
Jump to: navigation, search



  Index  



Nvidia-preferred-partner-badge-rgb-for-screen.png



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.



Previous: Deep Learning‎/TensorRT/Parsing Tensorflow Index Next: Deep Learning/TensorRT/Building Examples