冻结graph.pb到tflite,以便在Google Coral上运行(缺少输入和输出数组)

时间:2019-12-22 10:51:25

标签: python tensorflow google-colaboratory google-coral tf-lite

我已经创建了一个TensorFlow模型,其格式为Frozen graph.pb,我想在带有Google珊瑚的Rasberry Pi上运行它,所以我尝试使用以下代码对其进行转换:

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)

我在google collab上运行它,但不知道该如何用于输入和输出数组this is the model

0 个答案:

没有答案