Skip to main content

Parsing TensorFlow Lite Models (*.tflite) Easily

Project description

Easily Parse TFLite Models with Python

Build and Test Download

This tflite package parses TensorFlow Lite (TFLite) models (*.tflite), which are built by TFLite converter. For background, please refer to Introducing TFLite Parser Python Package.

Usage

Install the package and use it like what you build from the TensorFlow codebase. It's recommended to install the version that same as the TensorFlow that generates the TFLite model.

pip install tensorflow==2.3.0
pip install tflite==2.3.0

The raw API of tflite can be found in this documentation. The MobileNet test can serve as a usage example of parsing models.

Enhancements

The generated python package is not friendly to use sometimes. We have introduced several enhancements:

  • Easy import: A single import tflite (example) to replace importing every classes and funtions in tflite (example).
  • Builtin opcode helper: The opcode is encoded as digits which is hard to parse for human. Two APIs added to make it easy to use.

Compatibility Handling

TensorFlow sometimes leaves compability hanlding of the TFLite model to the users. As these are API breaking change that can be easily fixed, we do this in the tflite package.

Contributing Updates

As the operator definition may change across different TensorFlow versions, this package needs to be updated accordingly. If you notice that the package is out of date, please feel free to contribute new versions. This is pretty simple, instructions as below.

  1. Fork the repository, and download it.
  2. Install additional depdendency via pip install -r requirements.txt. And install flatbuffer compiler (you may need to manually build it).
  3. Generate the code for update. Tools have been prepared, there are prompt for actions.
    1. Download schema.fbs for a new version.
    2. Update the builtin operator mapping.
    3. Update the classes and functions import of submodules.
    4. Update the API document.
    5. Update the versioning in __init__.py.
    6. Build and test (simply pytest) around. Don't forget to re-install the newly built tflite package before testing it.
  4. Push your change and open Pull Request.
  5. The maintainer will take the responsibility to upload change to PyPI when merged.

Resources

License

Apache License Version 2.0 as TensorFlow's.

Disclaimer

The schema.fbs is obtained from TensorFlow directly. Maintainer of this package had tried to contact TensorFlow maintainers for licensing issues, but received no reply. Ownership or maintainship is open to transfer or close if there were any issue.

Project details


Download files

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

Source Distribution

tflite-2.18.0.tar.gz (49.2 kB view details)

Uploaded Source

Built Distribution

tflite-2.18.0-py2.py3-none-any.whl (153.2 kB view details)

Uploaded Python 2Python 3

File details

Details for the file tflite-2.18.0.tar.gz.

File metadata

  • Download URL: tflite-2.18.0.tar.gz
  • Upload date:
  • Size: 49.2 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.8

File hashes

Hashes for tflite-2.18.0.tar.gz
Algorithm Hash digest
SHA256 07e9478e0d0fb4d3082d0ce8e7e6eab4a90a7174412faec7702e24bfcbf4190d
MD5 fe88576aa4b9899bf8c7c06c7c8f7b3c
BLAKE2b-256 930e186beaed34ce9a740d7add374118b206fac75631c94bb4fa72d63ec32f7f

See more details on using hashes here.

File details

Details for the file tflite-2.18.0-py2.py3-none-any.whl.

File metadata

  • Download URL: tflite-2.18.0-py2.py3-none-any.whl
  • Upload date:
  • Size: 153.2 kB
  • Tags: Python 2, Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.8

File hashes

Hashes for tflite-2.18.0-py2.py3-none-any.whl
Algorithm Hash digest
SHA256 a49a4316530377c597bb272176c12d5d3c4453640118b068c93a8c3cc6c5c2c1
MD5 316229097a7c18f84e32bf2ecf755b7d
BLAKE2b-256 db9bf35d52989cb458a911f1ed4164b37dcf6559237b73c962c09ed0b6cecd4f

See more details on using hashes here.

Supported by

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