我看到网上的教程都写得太过于繁琐,不简洁,这也导致我当时找了一个晚上都没有成功转换.现在我将我的代码分享给大家,简单又好用.
我用的是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)
这里附上我自己测试的图片