Skip to content

This Unity package provides a foundation for performing inference with the Barracuda inference library. It includes a flexible base class to extend with task-specific packages.

Notifications You must be signed in to change notification settings

cj-mills/unity-barracuda-inference-base

Repository files navigation

Barracuda Inference Base

Barracuda Inference Base is a custom Unity package that provides a foundation for performing inference with the Barracuda inference library. It includes a flexible base class to extend with task-specific packages.

Features

  • Perform inference using the Unity Barracuda library
  • Extensible BarracudaModelRunner base class for custom inference tasks
  • Support for different compute backends and channel orders

Code Walkthrough

Extensions

GitHub Repository Description
unity-barracuda-inference-image-classification Perform image classification using computer vision models.
unity-barracuda-inference-yolox Perform object detection using YOLOX models.
unity-barracuda-inference-posenet Perform 2D human pose estimation using PoseNet models.

Demo Projects

GitHub Repository Description
barracuda-image-classification-demo Perform image classification using computer vision models.
barracuda-inference-yolox-demo Perform object detection using YOLOX models.
barracuda-inference-posenet-demo Perform 2D human pose estimation using PoseNet models.

Getting Started

Prerequisites

  • Unity game engine

Installation

You can install the Barracuda Inference Base package using the Unity Package Manager:

  1. Open your Unity project.
  2. Go to Window > Package Manager.
  3. Click the "+" button in the top left corner, and choose "Add package from git URL..."
  4. Enter the GitHub repository URL: https://github.com/cj-mills/unity-barracuda-model-runner.git
  5. Click "Add". The package will be added to your project.

For Unity versions older than 2021.1, add the Git URL to the manifest.json file in your project's Packages folder as a dependency:

{
  "dependencies": {
    "com.cj-mills.barracuda-model-runner": "https://github.com/cj-mills/unity-barracuda-model-runner.git",
    // other dependencies...
  }
}

License

This project is licensed under the MIT License. See the LICENSE file for details.

About

This Unity package provides a foundation for performing inference with the Barracuda inference library. It includes a flexible base class to extend with task-specific packages.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages