run_model

dragon.onnx.run_model(
  model,
  inputs,
  device='CUDA:0',
  **kwargs
)[source]

Execute an onnx model once.

Parameters:
  • model (str) – The path of onnx model file.
  • inputs (Union[Sequence, Dict]) – The input arrays.
  • device (str, optional) – The executing device.
Returns:

namedtuple – The model outputs.