用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)

这里附上我自己测试的图片
在这里插入图片描述

在这里插入图片描述

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值