pb模型转TFLite

最新推荐文章于 2022-09-02 20:54:19 发布

孤獨的心、

最新推荐文章于 2022-09-02 20:54:19 发布

阅读量327

点赞数

分类专栏：笔记文章标签： tensorflow 深度学习神经网络

本文链接：https://blog.csdn.net/qq_26174161/article/details/108210306

版权

笔记专栏收录该内容

5 篇文章 0 订阅

订阅专栏

import tensorflow as tf
saved_model_dir=''
inputs=["Placeholder"]
outputs=["final_result"]
converter = tf.lite.TFLiteConverter.from_frozen_graph('output_graph.pb',inputs,outputs)
#converter.optimizations = [tf.lite.Optimize.DEFAULT]
tflite_quant_model = converter.convert()
open("converted_model02.tflite", "wb").write(tflite_quant_model)