用于.tflite的tensorflow toco命令

Question

我正在遵循TFLite Android的步骤

我无法使用toco命令，因此无法生成："optimized_graph.lite“

下面是我使用的命令：

    IMAGE_SIZE=224
toco \
  --graph_def_file=tf_files/retrained_graph.pb \
  --output_file=tf_files/optimized_graph.lite \
  --input_format=TENSORFLOW_GRAPHDEF \
  --output_format=TFLITE \
  --input_shape=1,${IMAGE_SIZE},${IMAGE_SIZE},3 \
  --input_array=input \
  --output_array=final_result \
  --inference_type=FLOAT \
  --input_data_type=FLOAT

错误是：

F tensorflow/contrib/lite/toco/toco.cc:46]检查失败: parsed_toco_flags.input_file.specified()缺少必需的标志: input_file

在Mac OS上，我仔细遵循了说明，tensorflow的版本是1.7

Answer 1

当我跑的时候

toco --help

我在此命令的选项中看不到--graph_def_file。相反，我找到了--input_file，它的含义是输入文件(任何支持格式的模型)。对于Protobuf格式，无论文件扩展名如何，都支持文本和二进制。

在我用--input_file替换了--graph_def_file之后，这个命令起作用了，我得到了一个optimized_graph.lite文件。

下面是完整的命令。

toco \
  --input_file=tf_files/retrained_graph.pb \
  --output_file=tf_files/optimized_graph.lite \
  --input_format=TENSORFLOW_GRAPHDEF \
  --output_format=TFLITE \
  --input_shape=1,${IMAGE_SIZE},${IMAGE_SIZE},3 \
  --input_array=input \
  --output_array=final_result \
  --inference_type=FLOAT \
  --input_data_type=FLOAT