1. 首页
  2. 移动开发
  3. 其他
  4. keras .h5转移动端的.tflite文件实现方式

keras .h5转移动端的.tflite文件实现方式

上传者: 2020-11-29 16:34:57上传 PDF文件 50.24KB 热度 22次

以前tensorflow有bug 在winodws下无法转,但现在好像没有问题了,代码如下将keras 下的mobilenet_v2转成了tflite深度学习迅猛发展,目前已经可以移植到移动端使用了,TensorFlow推出的TensorFlow Lite就是一款把深度学习应用到移动端的框架技术。使用TensorFlowLite 需要tflite文件模型,这个模型可以由TensorFlow训练的模型转换而成。所以首先需要知道如何保存训练好的TensorFlow模型。checkpoints文件仅是保存训练好的权重,不带网络结构,所以做predict时需要结合model使用如:。于是需要自己定义一个relu6看到生成的tflite文件表示保存成功了也可以这么查看tflite网络的输入输出输出了以下信息

下载地址
用户评论