sh 从tensorflow冷冻模型到tensorflow lite

Posted 2021-05-06

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了sh 从tensorflow冷冻模型到tensorflow lite相关的知识，希望对你有一定的参考价值。

toco \
  --input_file=frozen_model.pb \
  --output_file=frozen_model.lite \
  --input_format=TENSORFLOW_GRAPHDEF \
  --output_format=TFLITE \
  --input_shape=1,100,100,3 \
  --input_array=input_image \
  --output_arrays=landmarks,gender,smile,glasses,head_pose \
  --inference_type=FLOAT \
  --input_data_type=FLOAT

以上是关于sh 从tensorflow冷冻模型到tensorflow lite的主要内容，如果未能解决你的问题，请参考以下文章

python 从冷冻模型到张量板

tensorflow如何查看模型的所有层的输入输出的tensor name

tensorflow 训练好的模型，怎么调用

tensorflow saver 保存和恢复指定 tensor

Tensorflow 类型错误：不允许使用 `tf.Tensor` 作为 Python `bool`。