12 lines
367 B
Python
Executable File
12 lines
367 B
Python
Executable File
#!/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)
|