Skip to main content

TFLite Model Maker: a model customization library for on-device applications.

Project description

TFLite Model Maker

Overview

The TFLite Model Maker library simplifies the process of adapting and converting a TensorFlow neural-network model to particular input data when deploying this model for on-device ML applications.

Requirements

  • Refer to requirements.txt for dependent libraries that're needed to use the library and run the demo code.

Installation

There are two ways to install Model Maker.

  • Install a prebuilt pip package.
pip install tflite-model-maker
  • Clone the source code from GitHub and install.
git clone https://github.com/tensorflow/examples
cd examples/tensorflow_examples/lite/model_maker/pip_package
pip install -e .

End-to-End Example

For instance, it could have an end-to-end image classification example that utilizes this library with just 4 lines of code, each of which representing one step of the overall process. For more detail, you could refer to Colab for image classification.

  1. Load input data specific to an on-device ML app.
data = ImageClassifierDataLoader.from_folder('flower_photos/')
  1. Customize the TensorFlow model.
model = image_classifier.create(data)
  1. Evaluate the model.
loss, accuracy = model.evaluate()
  1. Export to Tensorflow Lite model and label file in export_dir.
model.export(export_dir='/tmp/')

Notebook

Currently, we support image classification, text classification and question answer tasks. Meanwhile, we provide demo code for each of them in demo folder.

Project details


Release history Release notifications | RSS feed

Download files

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

Source Distribution

Built Distribution

File details

Details for the file tflite-model-maker-nightly-0.1.2.dev202009021601.tar.gz.

File metadata

  • Download URL: tflite-model-maker-nightly-0.1.2.dev202009021601.tar.gz
  • Upload date:
  • Size: 50.2 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/50.0.3 requests-toolbelt/0.9.1 tqdm/4.48.2 CPython/3.7.7

File hashes

Hashes for tflite-model-maker-nightly-0.1.2.dev202009021601.tar.gz
Algorithm Hash digest
SHA256 d95539b16bd71e0d119ddc321aaec3fb3d7f6d57200e67eb0254505f994c010f
MD5 79cffb5379f9996fbb53871542d721e6
BLAKE2b-256 65580bb72cd1216d7f3e8376e3c107d5cf680bb53a898ac610f1bfa4fe97274a

See more details on using hashes here.

File details

Details for the file tflite_model_maker_nightly-0.1.2.dev202009021601-py3-none-any.whl.

File metadata

File hashes

Hashes for tflite_model_maker_nightly-0.1.2.dev202009021601-py3-none-any.whl
Algorithm Hash digest
SHA256 9e83b989aab6cf5a84224d1d9f4133eea4b82e598f627ea8803962480df1fde2
MD5 8b2a95eee04e2ff91d3d436faa6b9f3e
BLAKE2b-256 49ddbb7b489a52fa1a2919b95335b8cdd26bee733ac4ede87f68b42d6cf6a89a

See more details on using hashes here.

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