Easily Parse TFLite Models with Python
tflite package parses TensorFlow Lite (TFLite) models (
*.tflite), which are built by TFLite converter. For background, please refer to Introducing TFLite Parser Python Package.
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.
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
- 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.
tflite.opcode2name(): get the type name of given opcode.
tflite.BUILTIN_OPCODE2NAME: a dict that maps the opcode to name of all the builtin operators.
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.OperatorCode.BuiltinCode(): maintains API compability in
2.4.0, see this issue.
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.
- Fork the repository, and download it.
- Install additional depdendency via
pip install -r requirements.txt. And install flatbuffer compiler (you may need to manually build it).
- Generate the code for update. Tools have been prepared, there are prompt for actions.
schema.fbsfor a new version.
- Update the builtin operator mapping.
- Update the classes and functions import of submodules.
- Update the API document.
- Update the versioning in
- Build and test (simply
pytest) around. Don’t forget to re-install the newly built
tflitepackage before testing it.
- Push your change and open Pull Request.
- The maintainer will take the responsibility to upload change to PyPI when merged.
Apache License Version 2.0 as TensorFlow’s.
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.