diff --git a/.gitignore b/.gitignore index ed07e02..8dbff6d 100644 --- a/.gitignore +++ b/.gitignore @@ -2,4 +2,4 @@ venv train.json dataset final_model - +final_model.tflite diff --git a/tfliteify.py b/tfliteify.py new file mode 100755 index 0000000..d366235 --- /dev/null +++ b/tfliteify.py @@ -0,0 +1,11 @@ +#!/usr/bin/env python3 + +import tensorflow as tf + +converter = tf.lite.TFLiteConverter.from_saved_model("final_model") +converter.target_spec.supported_ops = [ + tf.lite.OpsSet.TFLITE_BUILTINS, # enable TensorFlow Lite ops. + tf.lite.OpsSet.SELECT_TF_OPS # enable TensorFlow ops. +] +tflite_model = converter.convert() +open("final_model.tflite", "wb").write(tflite_model)