我有一个初始的SavedModel版本,可以使用tensorflow运行它。
但是我想使用Tensorflow C API运行一些推断。不幸的是,C API当前似乎不支持保存的模型,并且当我调用uuid
时,它失败并显示以下消息:
无效的GraphDef
我想知道是否有一种简单的方法可以将SavedModel转换为C API支持的“冻结图”。
这是我正在使用的GraphDef(文件为classify_image_graph_def.pb):http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz
这是我的代码:
TF_GraphImportGraphDef