用keras,在jupyter notebook中将.h5文件一步转换为.tflite文件(8行代码一步搞定)

我看到网上的教程都写得太过于繁琐,不简洁,这也导致我当时找了一个晚上都没有成功转换.现在我将我的代码分享给大家,简单又好用.

我用的是tensorflow2.4.0的版本

首先,先将代码附上

#在这里设置.h5文件的地址和.tflite的地址,.tflite的文件需要你手动创建
keras_model_address=""#.h5文件的地址
tflite_model_address=""#.tflite文件的地址
##############################
import tensorflow as tf
from keras import models

model =models.load_model(keras_model_address)
converter = tf.lite.TFLiteConverter.from_keras_model(model)

tflite_model = converter.convert()
open(tflite_model_address,"wb").write(tflite_model)

这里附上我自己测试的图片