Tensorflow Android:重新训练的初始v3需要花费太多时间

时间:2017-02-21 12:24:37

标签: android tensorflow

我用我自己的20个类别重新训练了Inception-V3最后一层。当我在Android演示应用程序中使用重新训练模型时,预测需要6到8秒。

上运行
  • LG G4手写笔 - > 6-8秒
  • S6, - > 3-4.5秒

我已完成optimize_for_inference需要6-9秒而quantize_graph需要7-11秒。有没有办法改善它?

LG G4手写笔输出:

Output

修改

我已关注this

1 个答案:

答案 0 :(得分:0)

将您的调试输出与我手机上的普通TF Classify应用进行比较我发现您有一个更大的节点数,这表明由于某种原因,您的图表比它应该大得多。我对量化方法不太熟悉,但看起来你的conv2D层数也比普通层多。

如果没有任何进一步的信息,我很难说,但我认为您应该重建图表并检查您是否已正确添加最终图层。