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.

pip install tflite-model-maker

If you want to install nightly version tflite-model-maker-nightly, please follow the command:

pip install tflite-model-maker-nightly
  • 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. Import the required modules.
from tflite_model_maker import image_classifier
from tflite_model_maker.image_classifier import DataLoader
    1. Load input data specific to an on-device ML app.
data = DataLoader.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.3.0.dev202104262251.tar.gz.

File metadata

  • Download URL: tflite-model-maker-nightly-0.3.0.dev202104262251.tar.gz
  • Upload date:
  • Size: 324.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.4.1 importlib_metadata/4.0.1 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.60.0 CPython/3.7.10

File hashes

Hashes for tflite-model-maker-nightly-0.3.0.dev202104262251.tar.gz
Algorithm Hash digest
SHA256 0208cf5d52c6682369664bce80ae40b0616a79a41ecf436cd6b02b68fe40ba24
MD5 e67a99c8019b4250def92f74d1ec45b6
BLAKE2b-256 d7c570d541503b41e031e9d0fa5568787e054c954d1b4e913f2f13640906ea20

See more details on using hashes here.

File details

Details for the file tflite_model_maker_nightly-0.3.0.dev202104262251-py3-none-any.whl.

File metadata

File hashes

Hashes for tflite_model_maker_nightly-0.3.0.dev202104262251-py3-none-any.whl
Algorithm Hash digest
SHA256 041582587b71040ae5be433f99403e00036ef7c3d4c6579fb31254d4cabe451a
MD5 244d885c87f3dbc9af168155c48fa6f7
BLAKE2b-256 4aec5186a943cee70d4c02de5550e432e625444f968824ab09624bb5382f3659

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