keras模型转tensorflow lite

import tensorflow as tf
model=tf.keras.models.load_model('models/cnn.h5')
# 转换模型
converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_model = converter.convert()
# Save the model.
with open('model.tflite', 'wb') as f:
  f.write(tflite_model)