Import graph

Arm NN provides parsers for reading model files from neural network frameworks. There are typically two steps to do this:

  1. Load the model.
  2. Bind the input and output points of its graph.

Code to import the TensorFlow model

TensorFlow graphs in both text and binary ProtoBuf formats are supported. For more details on freezing TensorFlow graphs to include their weights, see this guide.

Tip: After this step, the code is common regardless of the framework that you started with as the INetwork and two BindingPointInfo objects provide everything that is needed.

Previous Next