get_onnx_output_infos

get_onnx_output_infos(model: str | Path | onnx.ModelProto) -> dict[str, dict[str, Any]]

• Description: Returns output shape and dtype info of an ONNX model.

• Dependencies

  • Requires onnx.

• Example

  from capybara.onnxengine import get_onnx_output_infos
  
  infos = get_onnx_output_infos("model.onnx")
  print(infos)