【问题标题】:TensorFlow generate frozen pb model to use on AndroidTensorFlow 生成冻结 pb 模型以在 Android 上使用
【发布时间】:2018-04-28 23:03:17
【问题描述】:

我一直在尝试使用 Object Detection API (https://github.com/tensorflow/models/blob/master/research/object_detection/export_inference_graph.py) 提供的方法生成冻结模型以用于 Android 的 TF Detect,但出现以下错误:

E/AndroidRuntime: Caused by: java.io.IOException: Not a valid 
TensorFlow Graph serialization: NodeDef mentions attr 'T' not in 
Op<name=Where; signature=input:bool -> index:int64>; NodeDef: 

显然 pb 一代对 Android 无效。但是,我确实使用 Google 提供的 pb 模型 (http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_android_export.zip) 对此进行了测试,它适用于 Android。

我的问题是,有没有一种特殊的方法可以为 Android 生成 pb 模型?

【问题讨论】:

    标签: tensorflow


    【解决方案1】:

    当我使用量化的 ssd_mobilenet_v1 模型时,我也遇到了这个问题。

    【讨论】:

    • 你解决了吗?我在官方 repo 上打开了一个 issue,我收到了一些 cmets,但是没有一个起作用;这就是问题所在:github.com/tensorflow/models/issues/2823我仍在努力解决问题,如果我设法找到解决方案,我会发布它。
    猜你喜欢
    • 2018-01-08
    • 2019-01-04
    • 1970-01-01
    • 1970-01-01
    • 2018-07-03
    • 1970-01-01
    • 2022-01-19
    • 2020-03-06
    • 1970-01-01
    相关资源
    最近更新 更多