移动设备上的 TensorFlow lite 精度下降答案

【问题标题】：Tensorflow lite accuracy drop on mobile device移动设备上的 TensorFlow lite 精度下降
【发布时间】：2019-01-08 01:58:34
【问题描述】：

我遵循了 TensorFlow for Poets 教程： Tensorflow for Poets 1 和 Tensorflow for Poets 2。我重新训练的模型在我的笔记本电脑上进行测试时给出了准确的结果，但在转换为 .tflite 文件并尝试在我的 Android 设备上对同一图像进行分类后，准确度下降到 1% 以下。

我使用以下命令重新训练和转换：

  python retrain.py \
  --bottleneck_dir=tf_files/bottlenecks \
  --how_many_training_steps=500 \
  --model_dir=tf_files/models/ \
  --summaries_dir=tf_files/training_summaries/"${ARCHITECTURE}" \
  --output_graph=tf_files/retrained_graph.pb \
  --output_labels=tf_files/retrained_labels.txt \
  --architecture="${ARCHITECTURE}" \
  --image_dir=tf_files/flower_photos

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,224,224,3 \
  --input_array=Placeholder \
  --output_array=final_result \
  --inference_type=FLOAT \
  --input_data_type=FLOAT

奇怪的是，优化后的文件几乎和原始文件一样高（都在 80 MB 左右）。

使用 TensorFlow 1.9.0 和 Python 3.6.6。

感谢任何帮助或提示！

【问题讨论】：

这里有答案，看看这个：stackoverflow.com/a/58583602/11517841

标签： android python tensorflow

【解决方案1】：

嗯，我想通了。显然 ARCHITECTURE 变量未设置为正确的值。所以如果有人遇到同样的问题，首先检查一下

【讨论】：