Inspect a model

Get model inputs and outputs

Use the inputs property of the runtime model to get the inputs of the model, and the name or tensor shape of each input.

For example:

using UnityEngine;
using System.Collections.Generic;
using Unity.Sentis;

public class GetModelInputs : MonoBehaviour
{

    public ModelAsset modelAsset;

    void Start()
    {
        Model runtimeModel = ModelLoader.Load(modelAsset);

        List<Model.Input> inputs = runtimeModel.inputs;

        // Loop through each input
        foreach (var input in inputs)
        {
            // Log the name of the input, for example Input3
            Debug.Log(input.name);

            // Log the tensor shape of the input, for example (1, 1, 28, 28)
            Debug.Log(input.shape);

        }
    }
}

Input dimensions are fixed or dynamic. Refer to Model inputs for more information.

Use the outputs property of the runtime model to get the names of the output layers of the model.

For example:

        List<string> outputs = runtimeModel.outputs;

        // Loop through each output
        foreach (var output in outputs)
        {
            // Log the name of the output
            Debug.Log(output);
        }

Get layers and layer properties

Use the layers property of the runtime model to get the neural network layers in the model, and the name, inputs, outputs or flags of each layer.

Open a model as a graph

To open a model as a graph, follow these steps:

  1. Install Netron, a third-party viewer for neural networks.
  2. Double-click a model asset in the Project window, or select the model asset then select Open in the Model Asset Import Settings window.

Additional resources