【发布时间】: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