将Tensorflow对象检测Api模型转换为TFLite

时间:2019-08-05 08:41:24

标签: python-2.7 tensorflow object-detection-api tensorflow-lite

我使用对象检测Api中的provided documentation用Tensorflow训练了SSDLite-MobilenetV2模型。然后,我通过运行export_tflite_ssd_graph脚本导出模型。生成了一个pb和一个pbtxt文件。最后,我尝试使用tflite_convert命令将模型转换为tflite格式。但是,出现以下错误:

回溯(最近通话最近): 在第11行的文件“ / usr / local / bin / tflite_convert” sys.exit(main()) 主目录中的文件“ /usr/local/lib/python2.7/dist-packages/tensorflow/contrib/lite/python/tflite_convert.py”,第412行 app.run(main = run_main,argv = sys.argv [:1]) 运行中的文件“ /usr/local/lib/python2.7/dist-packages/tensorflow/python/platform/app.py”,第125行 _sys.exit(main(argv)) 在run_main中的第408行,文件“ /usr/local/lib/python2.7/dist-packages/tensorflow/contrib/lite/python/tflite_convert.py” _convert_model(tflite_flags) _convert_model中的第100行,文件“ /usr/local/lib/python2.7/dist-packages/tensorflow/contrib/lite/python/tflite_convert.py” 转换器= _get_toco_converter(标志) _get_toco_converter中的文件“ /usr/local/lib/python2.7/dist-packages/tensorflow/contrib/lite/python/tflite_convert.py”,第87行 返回converter_fn(** converter_kwargs) 文件“ /usr/local/lib/python2.7/dist-packages/tensorflow/contrib/lite/python/lite.py”,行340,位于from_saved_model中 output_arrays,tag_set,signature_key) 文件“ /usr/local/lib/python2.7/dist-packages/tensorflow/contrib/lite/python/convert_saved_model.py”,第239行,在freeze_saved_model中 meta_graph = get_meta_graph_def(saved_model_dir,tag_set) 在get_meta_graph_def的第61行,文件“ /usr/local/lib/python2.7/dist-packages/tensorflow/contrib/lite/python/convert_saved_model.py” 返回loader.load(sess,tag_set,Saved_model_dir) 载入中的文件“ /usr/local/lib/python2.7/dist-packages/tensorflow/python/saved_model/loader_impl.py”,第197行 返回loader.load(sess,tags,import_scope,** saver_kwargs) 载入中的文件“ /usr/local/lib/python2.7/dist-packages/tensorflow/python/saved_model/loader_impl.py”,第350行 ** saver_kwargs) 在load_graph中的第275行,文件“ /usr/local/lib/python2.7/dist-packages/tensorflow/python/saved_model/loader_impl.py” meta_graph_def = self.get_meta_graph_def_from_tags(标签) 在get_meta_graph_def_from_tags中的第251行,文件“ /usr/local/lib/python2.7/dist-packages/tensorflow/python/saved_model/loader_impl.py” “在SavedModel中找不到。要检查其中的可用标记集” RuntimeError:在SavedModel中找不到与标签set(['serve'])关联的MetaGraphDef。要检查SavedModel中可用的标签集,请使用SavedModel CLI:saved_model_cli

似乎转换脚本不包含SERVING标签常量。我该如何解决?

我正在使用tensorflow-gpu 1.12.0

0 个答案:

没有答案