s4k10503 / onnx-operators-and-activations-checker-for-barracuda

Check for operators and activation functions that are not supported by Unity Barracuda.

ONNX Operators and Activations Checker for Barracuda

Overview

This application allows users to load ONNX models and automatically check for operators and activation functions that are not supported by Unity Barracuda.
It is built using Python and leverages the customtkinter library for a modern and stylish GUI.

Features

• Load ONNX Model: A user-friendly GUI button for loading ONNX files.
• Check Operators: Automatically scans the ONNX model to identify unsupported operators.
• Check Activation Functions: Automatically scans the ONNX model to identify unsupported activation functions.
• Progress Bar: Provides real-time feedback during the checking process.

Dependencies

• Python 3.x
• ONNX
• customtkinter
• tkinter

Running with Docker

To run this application using Docker, you can build and run a Docker container as follows:

Build the Docker Image

Run the following command to build the Docker image:

docker build -t onnx_checker .

Run the Docker Container

Run the following command to run the Docker container.
This also mounts the host directory to the container, enabling file access for the GUI.

docker run -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix -v /path/on/host:/path/in/container onnx_checker

Note: Replace /path/on/host and /path/in/container with the appropriate paths.

Security Note: Make sure to run xhost + on the host machine to allow connections to the X server:

xhost +

After you are done, restrict the X server access again with:

xhost -

License

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

onnx-operators-and-activations-checker-for-barracuda's Issues

Inject Dependencies

I want to inject dependencies using a DI library.
It is assumed that a DI container will be realized using the injector module.