- Fix the requirement deps issues.
- Fix the version of tflite-support and flatbuffers to be consistent with other packages.
- Fix the version of scann to 1.2.6 to solve the import failure.
- Add one new task: Searcher that support Image / Text as the inputs.
- Minor fix: remove tensorflow-estimator version requirements to avoid the incompatible version error message during pip installation.
- Minor fix: update tensorflow==2.6 and fix the failure of
evaluate_tflite
for the object detection task.
- Minor fix: Move librosa to an optional library for audio task.
- Polish document showing in https://www.tensorflow.org/lite/api_docs/python/tflite_model_maker.
- Add some 0.2.x APIs for backward compatibility.
- Export TFLite Model Maker public APIs, and adapt colabs as demo code.
- Add three new tasks: Object Detection, Audio Classification and Recommendation.
- Refactor the code to be more flexible and extentable without changing API.
- Add executable test script for pip package.
- Update tensorflow-hub version in requirements.txt to fix incompatible error when installing pip package.
- Update stable tensorflow version in requirements.txt.
- Support exporting to TFJS models.
- Refine Code: refactor the
gen_dataset
internal API inDataLoader
and remove redundant methods. - Split requirements to stable version and nightly version.
- Fix a bug for missing
default_batch_size
. - Add a demo of customized task.
- Refine Code: Refactor
model_spec
to split different model specifications into image/text and addClassificationDataLoader
.
- Update tf-nightly version in requirements.txt.
- Refine Code: Change pre-defined model specification to function and extract Keras callback logic into a reusable function.
- Set the required package version in requirements.txt.
- Refine Code: Move
load_from_tfds
intoImageDataLoader
, refactor theDataLoader
code, unify the predict method and provide a default implementation for TFLite expo.
- Setup virtualenv and test pip package.
Initial version for TFLite Model Maker: a model customization library for on-device applications.
Support tasks:
- Add
metadata_writer
for nlp tasks and setwith_metadata=True
by default when exporting to TFLite. - Update
model_info
in metadata for image classifier.
- Update README.
- Add
mobilebert_qa_squad
model_spec, a mobilebert model pre-trained with Squad1.1 dataset.
- Initial version.