【问题标题】:Tensorflow Android : retrained Inception v3 take too much timeTensorflow Android:重新训练的 Inception v3 花费了太多时间
【发布时间】:2017-02-21 12:24:37
【问题描述】:

我用我自己的 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

【问题讨论】:

    标签: android tensorflow


    【解决方案1】:

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

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

    【讨论】:

    • @JCookr 我尝试重新训练图表并应用优化,但得到了相同的结果。
    • 我认为您不仅需要重新训练,还需要重新构建形状。你在看教程吗?
    • 是的,我关注了thisthis
    • 你能指导我更多吗?
    • 我不确定我还能提供多少帮助。我可以建议您在 Tensorboard 中可视化您的图表并确保它是您所期望的。将其与标准初始图进行比较可能是一个不错的起点。
    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2017-03-28
    • 1970-01-01
    • 1970-01-01
    • 2017-10-24
    • 2020-05-19
    相关资源
    最近更新 更多