tensorflow 2 model to tflite code example

Example: how to convert tensorflow 1.15 model to tflite

# Converting a GraphDef from session.converter = tf.compat.v1.lite.TFLiteConverter.from_session(  sess, in_tensors, out_tensors)tflite_model = converter.convert()open("converted_model.tflite", "wb").write(tflite_model)# Converting a GraphDef from file.converter = tf.compat.v1.lite.TFLiteConverter.from_frozen_graph(  graph_def_file, input_arrays, output_arrays)tflite_model = converter.convert()open("converted_model.tflite", "wb").write(tflite_model)# Converting a SavedModel.converter = tf.compat.v1.lite.TFLiteConverter.from_saved_model(    saved_model_dir)tflite_model = converter.convert()open("converted_model.tflite", "wb").write(tflite_model)# Converting a tf.keras model.converter = tf.compat.v1.lite.TFLiteConverter.from_keras_model_file(    keras_model)tflite_model = converter.convert()open("converted_model.tflite", "wb").write(tflite_model)

Tags:

Misc Example