asus4 / onnxruntime-unity

ONNX Runtime Plugin for Unity
54 stars 2 forks source link
onnxruntime unity

ONNX Runtime Plugin for Unity

upm

Pre-built ONNX Runtime libraries for Unity.

See Examples

https://github.com/asus4/onnxruntime-unity-examples

Yolox

https://github.com/asus4/onnxruntime-unity-examples/assets/357497/96ed9913-41b7-401d-a634-f0e2de4fc3c7

NanoSAM

https://github.com/asus4/onnxruntime-unity-examples/assets/357497/5e2b8712-87cc-4a3a-82b7-f217087a0ed1

Tested environment

Execution Providers & Extensions

Execution Providers

Execution Providers are hardware acceleration libraries for each platform. See official docs for more details.

Platform CPU CoreML NNAPI CUDA TensorRT DirectML XNNPACK
macOS :white_check_mark: :white_check_mark:
iOS :white_check_mark: :white_check_mark: :construction:
Android :white_check_mark: :white_check_mark: :construction:
Windows :white_check_mark: :construction: :construction: :white_check_mark:
Linux :white_check_mark: :construction: :construction:

ONNX Runtime Extensions

ONNX Runtime Extensions are a set of pre/post-processing.

Platform Extensions
macOS :construction:
iOS :construction:
Android :construction:
Windows :construction:
Linux :construction:

:white_check_mark: : Supported in Unity Core library
:construction: : Experimental Preview

How to Install

Pre-built libraries are available on NPM. Add the following scopedRegistries and dependencies in Packages/manifest.json.

  "scopedRegistries": [
    {
      "name": "NPM",
      "url": "https://registry.npmjs.com",
      "scopes": [
        "com.github.asus4"
      ]
    }
  ]
  "dependencies": {
    "com.github.asus4.onnxruntime": "0.2.1",
    "com.github.asus4.onnxruntime.unity": "0.2.1",
    "com.github.asus4.onnxruntime.win-x64-gpu": "0.2.1",
    "com.github.asus4.onnxruntime.linux-x64-gpu": "0.2.1",
    "com.github.asus4.onnxruntime-extensions": "0.2.1",
    ... other dependencies
  }

What is included in each package