keras转tensorflow lite【方法二】直接转：简单模型实例

ゝ一纸荒年。 2022-03-29 09:52 242阅读 0赞

在youtube看到一个很赞的视频，直接将简单keras模型转tensorflow lite，中间不经过转tensorflow的过程。

之后有空会贴出具体实现图示。现在只贴出链接和代码。

链接：[https://www.youtube.com/watch?v=MZx1fhbL2q4][https_www.youtube.com_watch_v_MZx1fhbL2q4]

代码：

import tensorflow as tf
    import numpy as np
    from tensorflow import keras
    from tensorflow.contrib import lite
    
    model = keras.Sequential([keras.layers.Dense(units=1, input_shape=[1])])
    model.compile(optimizer='sgd', loss='mean_squared_error')
    
    xs = np.array([-1.0, 0.0, 1.0, 2.0, 3.0, 4.0], dtype=float)
    ys = np.array([-3.0, -1.0, 0.0, 3.0, 5.0, 7.0], dtype=float)
    
    model.fit(xs, ys, epochs=500)
    
    print(model.predict([10.0]))
    
    keras_file = "linear.h5"
    keras.models.save_model(model, keras_file)
    
    #converter = lite.TocoConverter.from_keras_model_file(keras_file)
    converter = lite.TFLiteConverter.from_keras_model_file(keras_file)
    tflite_model = converter.convert()
    open("linear.tflite", "wb").write(tflite_model)

环境：linux

图示：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTc3MDE2OQ_size_16_color_FFFFFF_t_70][]

[https_www.youtube.com_watch_v_MZx1fhbL2q4]: https://www.youtube.com/watch?v=MZx1fhbL2q4
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTc3MDE2OQ_size_16_color_FFFFFF_t_70]: /images/20220329/5ff547f1882e47b7883f6ebdee493b93.png

发表评论取消回复

表情：

评论列表（有 0 条评论，242人围观）

还没有评论，来说两句吧...

相关阅读

相关转换pb_Keras模型转换成Tensorflow模型方法

![46750eba0315531fd2232ac2cf381157.png][] 由于生产需要、项目组需要，需要将之前神经网络训练的模型（keras模型hdf5类型），转换

我会带着你远行/ 2023年01月07日 10:22/ 0 赞/ 243 阅读

相关 Keras tensorflow Resnet模型

Resnet相关文章整理 1、Keras大法（9）——实现ResNet-34模型 [https://blog.csdn.net/weixin\_42499236/a

小灰灰/ 2022年12月27日 07:26/ 0 赞/ 276 阅读

相关 tensorflow2.0+keras简单实现BERT模型

文章目录 BERT模型简介 BERT模型拆解 multi-head attention FeedFor

迈不过友情╰/ 2022年11月01日 13:42/ 0 赞/ 287 阅读

相关 tensorflow转onnx模型

一、准备相应的库 tensorflow当然是要的。试了下 python:3.6.13 tensorflow:1.14.0 python 3.7.

朱雀/ 2022年10月31日 00:47/ 0 赞/ 368 阅读

相关【Tensorflow+Keras】用Tensorflow.keras的方法替代keras.layers.merge

1 旧版在旧版的keras中，keras.layers.merge的使用方法 keras.layers.merge([layer1,layer2],

柔光的暖阳◎/ 2022年10月05日 14:56/ 0 赞/ 212 阅读

相关 keras转tensorflow

参考：[https://juejin.im/post/5b7bccc6e51d453887102e0a][https_juejin.im_post_5b7bccc6e51d45

野性酷女/ 2022年04月10日 13:16/ 0 赞/ 233 阅读

相关 keras转tensorflow lite【方法一】2步走

本文介绍的将keras转tensorflow lite的方法，共分为两步： keras转tensorflow，再将tensorflow转tensorflow lite

叁歲伎倆/ 2022年03月29日 09:52/ 0 赞/ 340 阅读

相关 keras转tensorflow lite【方法二】直接转：简单模型实例

在youtube看到一个很赞的视频，直接将简单keras模型转tensorflow lite，中间不经过转tensorflow的过程。之后有空会贴出具体实现图示。现在只贴出

ゝ一纸荒年。/ 2022年03月29日 09:52/ 0 赞/ 243 阅读

相关 keras转tf并加速（1）Keras转TensorFlow，并调用转换后模型进行预测

由于方便快捷，所以先使用Keras来搭建网络并进行训练，得到比较好的模型后，这时候就该考虑做成服务使用的问题了，TensorFlow的serving就很合适，所以需要把Kera

待我称王封你为后i/ 2022年02月01日 23:57/ 0 赞/ 358 阅读

相关 tensorflow中ckpt转pb模型

参考代码： import tensorflow as tf import os from tensorflow.python.framework import gra

朱雀/ 2022年01月22日 01:17/ 0 赞/ 251 阅读