Skip to main content

Tool to check if a ML model is compatible with the TOSA specification

Project description

TOSA Checker

Overview

The TOSA Checker is a tool that provides an easy way to ensure that a TensorFlow™ Lite model is compatible with the TOSA specification.

The project is currently in alpha, some stability issues may still be found.

Prerequisites

To build the TOSA Checker the following are required:

  • Ubuntu® 20.04.03 LTS (the TOSA Checker has been tested on this OS, but should work on other Linux® OS)
  • Bazel™ (Bazelisk is an easy way to install the correct version of Bazel™)
  • Git™
  • Python® >=3.8 (python, python-dev and python-pip)

Building and installation

pip install -U pip setuptools wheel
pip install -U numpy GitPython
python setup.py bdist_wheel
pip install dist/<tosa_checker>.whl

Docker™ image

A Docker™ image to build the TOSA Checker is also provided. More information can be found on docker/README.md.

Usage

>>> import tosa_checker as tc
>>> checker = tc.TOSAChecker(model_path="model.tflite")
>>> checker.is_tosa_compatible()
True

Security

To report any security issue in the TOSA Checker, please see SECURITY.md.

License

TOSA Checker is licensed under [Apache License 2.0].

Trademarks and Copyrights

  • Bazel™ is a trademark of Google® LLC.
  • Git™ is a trademark of Software Freedom Conservancy.
  • Linux® is the registered trademark of Linus Torvalds in the U.S. and elsewhere.
  • Python® is a registered trademark of the PSF.
  • Ubuntu® is a registered trademark of Canonical.
  • Tensorflow™ is a trademark of Google® LLC.
  • Docker™ is a trademark of Docker, Inc.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release. See tutorial on generating distribution archives.

Built Distributions

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page