Hugging Face's logo

unity
/
inference-engine-mobilenet-v2

Image Classification
unity-sentis
ONNX
unity-inference-engine
Model card Files
xet
 Community
inference-engine-mobilenet-v2 / README.md
UnityGiles's picture
UnityGiles
update to inference engine
d6bd9e7
preview code
|
raw
history blame contribute delete
1.04 kB
metadata 
license: mit
library_name: unity-sentis
pipeline_tag: image-classification
tags:
  - unity-inference-engine

MobileNet V2 in Unity 6 with Inference Engine

This is the MobileNet V2 model running in Unity 6 with Inference Engine. This is a small image classification model.

How to Use

  • Create a new scene in Unity 6;
  • Install com.unity.ai.inference from the package manager;
  • Add the RunMobileNet.cs script to the Main Camera;
  • Drag the mobilenet_v2.onnx asset from the models folder into the Model Asset field;
  • Drag the class_desc.txt asset from the data folder into the Labels Asset field;
  • Drag an image, such as Bee.jpg asset from the images folder into the Input Image field;

Preview

Enter play mode. If working correctly the predicted class will be logged to the console.

Inference Engine

Inference Engine is a neural network inference library for Unity. Find out more here.