我不明白如何在从GitHub提取的项目中从Tensorflow切换到Tensorflow Lite

Question

我正在尝试根据从GitHub（https://github.com/vanhuyz/CycleGAN-TensorFlow）上获取的CycleGAN创建.tflite模型。

我在这个领域很新，不了解如何在.tflite模型中公开.pb模型（我已经从检查点创建了该模型）。

我尝试使用tflite_convert但没有任何结果，也是因为我不知道要插入的参数为--input_arrays和--output_arrays。

有什么主意吗？

Answer 1

我建议在https://www.tensorflow.org/lite/convert/python_api处使用TFLiteConverter python api，并使用SavedModel作为模型输入格式。否则，您可以提供输入和输出张量名称或您的pb模型作为input_arrays和output_arrays。