TensorFlow生成在Android上使用的冻结pb模型

时间:2017-11-15 16:10:23

标签: tensorflow

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

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模型?

1 个答案:

答案 0 :(得分:0)

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