unity/inference-engine-mobilenet-v2

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.