TF对象检测API - 在冻结和量化我精心调整的网络后,无法运行量化网络

时间:2017-07-04 11:58:28

标签: tensorflow

TensorFlow对象检测API

使用TensorFlow对象检测API在我自己的DataSet上重新训练MobileNet。当我尝试运行已经冻结和量化的推理图时,会出现问题。

系统:

  • Ubuntu 16.04,
  • TensorFlow 1.2(来自源,仅限CPU),
  • Bazel 0.4.5

问题:

  • 使用模型动物园提供的frozen_graph.pb。
  • 使用量化为8位 巴泽勒滨/ tensorflow /工具/ graph_transforms / transform_graph。
  • 运行推理

然而,这是有效的

  • 使用object_detection / export_inference_graph.py重新训练并制作我自己的frozen_graph.pb
  • 使用bazel-bin / tensorflow / tools / graph_transforms / transform_graph
  • 量化为8位
  • 运行推理< - 生成错误

不起作用,我在尝试运行图表时遇到的错误是:

  

文件   " /home/unibap/TensorFlow/tensorflow-python2-sse4.2/local/lib/python2.7/site-packages/tensorflow/python/client/session.py" ;,   第1298行,在_do_call中       raise type(e)(node_def,op,message)tensorflow.python.framework.errors_impl.InvalidArgumentError:节点   '预处理器/图/同时/ ResizeImage / ResizeBilinear / eightbit'具有   来自不同帧的输入。输入   '预处理器/图/同时/ ResizeImage /大小'在框架内   '预处理器/图/同时/预处理器/图/同时/&#39 ;.输入   '预处理器/图/同时/ ResizeImage / ResizeBilinear_eightbit /预处理器/图/同时/ ResizeImage / ExpandDims /量化'   在框架中'

由于我可以量化并运行提供的frozen_graph.pb,问题必须出在导出工具上?使用哪个导出工具来创建模型动物园中的frozen_graph.pb?或者导出工具是如何调用的?

PS: 引用export_inference_graph.pb中的注释,确保在提供检查点时它应该生成一个冻结图。

"可选地,可以通过转换所提供的权重来冻结图形 检查点作为图形常量,从而消除了使用检查点的需要 推理过程中的文件。"

最佳

0 个答案:

没有答案