getModelInfo method

ModelInfo getModelInfo(
  1. Interpreter interpreter
)

Returns info about a loaded model's input/output tensors.

Implementation

ModelInfo getModelInfo(Interpreter interpreter) {
  final inputTensors = interpreter.getInputTensors();
  final outputTensors = interpreter.getOutputTensors();

  return ModelInfo(
    inputShapes: inputTensors.map((t) => t.shape).toList(),
    outputShapes: outputTensors.map((t) => t.shape).toList(),
    inputTypes: inputTensors.map((t) => t.type).toList(),
    outputTypes: outputTensors.map((t) => t.type).toList(),
  );
}