Skip to content

This tutorial covers creating an object detection plugin for a Unity game engine project using ONNX Runtime and DirectML.

License

Notifications You must be signed in to change notification settings

cj-mills/onnx-directml-unity-tutorial

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ONNX Runtime → DirectML → Unity → Tutorial

Tutorial Links

  • Training Tutorial: Train a YOLOX model using IceVision.
  • Part 1: Create a dynamic link library (DLL) file in Visual Studio to perform object detection with a YOLOX model using ONNX Runtime and DirectML.
  • Part 2: Perform object detection in a Unity project with ONNX Runtime and DirectML.

Demo Video

unity-onnx-directml-demo.mp4

Training Code

Jupyter Notebook Colab         Kaggle        
GitHub Repository Open In Colab Kaggle

Note: Training on the free GPU tier for Google Colab takes approximately 11 minutes per epoch, while training on the free GPU tier for Kaggle Notebooks takes around 15 minutes per epoch.

Kaggle Datasets

Reference Images


Class Image
call call
dislike dislike
fist  fist
four four
like  like
mute  mute
ok  ok
one  one
palm  palm
peace peace
peace_inverted peace_inverted
rock rock
stop stop
stop_inverted stop_inverted
three three
three2 three2
two_up  two_up
two_up_inverted two_up_inverted

About

This tutorial covers creating an object detection plugin for a Unity game engine project using ONNX Runtime and DirectML.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published