且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

怎么把.pb转换成.tflite?

更新时间:2023-02-19 14:21:34

使用以下代码时,无需指定输入和输出数组:

No need to specify input and output array, when using the following code:

import tensorflow as tf

converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

尝试