时间戳

首页 > 解决方案 > 将 Tensorflow 模型转换为 Tensorflow Lite

python - 将 Tensorflow 模型转换为 Tensorflow Lite

问题描述

我在将 tensorflow 模型转换为 tflite 时遇到问题。我有一个基于Tensorflow 对象检测的学习模型 我想使用TFlite 转换器的转换代码

curl https://storage.googleapis.com/download.tensorflow.org/models/mobilenet_v1_0.50_128_frozen.tgz | tar xzv -C /tmp

tflite_convert \
 --output_file=/tmp/foo.tflite \
 --graph_def_file=/tmp/mobilenet_v1_0.50_128/frozen_graph.pb \
 --input_arrays=input \
 --output_arrays=MobilenetV1/Predictions/Reshape_1

我不知道从哪里获得 input_arrays 和 output_arrays 的值。

感谢您的回答

标签： pythontensorflowtensorflow-lite

解决方案

我发现最简单的方法是将我的模型上传到https://lutzroeder.github.io/netron/它显示任何模型的输入和输出。

ps 它还有一个我使用的桌面应用程序，而不是上传每个模型

推荐阅读