UnityGiles's picture
update to inference engine
d6bd9e7
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.