跳转到主要内容

检查机器学习模型是否与TOSA规范兼容的工具

项目描述

TOSA Checker

概述

TOSA Checker是一个工具,它提供了一种简单的方法来确保TensorFlow™ Lite模型与TOSA规范兼容。

该项目目前处于alpha阶段,可能仍然存在一些稳定性问题。

先决条件

要构建TOSA Checker,需要以下内容

  • Ubuntu® 20.04.03 LTS(TOSA Checker已在此操作系统上进行了测试,但应适用于其他Linux®操作系统)
  • Bazel™(Bazelisk是安装正确版本的Bazel™的简单方法)
  • Git™
  • Python® >=3.8(python, python-dev和python-pip)

构建和安装

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

Docker™镜像

还提供了一个Docker™镜像来构建TOSA Checker。更多信息可以在docker/README.md中找到。

使用方法

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

安全性

如需报告TOSA Checker中的任何安全问题,请参阅SECURITY.md

许可证

TOSA Checker遵循[Apache License 2.0]许可。

商标和版权

  • Bazel™是Google® LLC的商标。
  • Git™是Software Freedom Conservancy的商标。
  • Linux®是美国及其他地区的注册商标,商标所有人为Linus Torvalds。
  • Python®是PSF的注册商标。
  • Ubuntu®是Canonical的注册商标。
  • Tensorflow™是Google® LLC的商标。
  • Docker™是Docker, Inc的商标。

项目详情


下载文件

下载适合您平台的应用程序文件。如果您不确定该选择哪个,请了解有关安装包的更多信息。

源代码分发

此版本没有可用的源代码分发文件。请参阅生成分发归档教程

构建分发

tosa_checker-0.2.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (16.7 MB 查看哈希值)

上传时间 CPython 3.11 manylinux: glibc 2.17+ x86-64

tosa_checker-0.2.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (16.7 MB 查看哈希值)

上传时间 CPython 3.10 manylinux: glibc 2.17+ x86-64

tosa_checker-0.2.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (16.7 MB 查看哈希值)

上传时间 CPython 3.9 manylinux: glibc 2.17+ x86-64

tosa_checker-0.2.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (16.7 MB 查看哈希值)

上传时间 CPython 3.8 manylinux: glibc 2.17+ x86-64

由...