我一直在尝试使用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模型?
答案 0 :(得分:0)
当我使用量化的ssd_mobilenet_v1模型时,我也遇到了这个问题。