有了pb文件,那我们必须把它转化为tf文件啊
import tensorflow as tf
graph_def_file = "E:/mian/tflllll/my_frozen_graph.pb"
input_arrays = ["Reshape_1"]
output_arrays = ["add_2"]
converter = tf.lite.TFLiteConverter.from_frozen_graph(
graph_def_file, input_arrays, output_arrays)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)
input_arrays选择前几个输入节点会报错,大概是形状未知?
博客提到有了pb文件需将其转化为tf文件,在选择input_arrays前几个输入节点时会报错,原因可能是形状未知。
1404

被折叠的 条评论
为什么被折叠?



