定制的Frozen_inference_graph.pb无法正常工作？

Question

我正在研究车牌检测软件。我在Google合作实验室中针对定制车牌数据集训练了TensorFlow对象检测API。之后，我从训练文件夹中下载了Frozen_inference_graph.pb，这是训练后的最终结果。最初，我使用了从TensorFlow集线器下载的save_model.pb，它可以正常工作，但现在我将此Frozen_graph_inference.pb重命名为saved_model.pb，使用时给我一个错误

我用于加载SAVED_MODEL的代码

#picking an object detection module

module_handle =“模型/” 检测器= hub.load（module_handle）.signatures ['default']

这里的模型是我存储所有模型的目录。 当我使用定制模型时，会出现此错误

文件“ objectDetection.py”，第2行，在 从helperFunction导入* 文件“ C：\ MachineLearning \ helperFunctions.py”，第89行，在 检测器= hub.load（module_handle）.signatures ['default'] getitem 中的文件“ C：\ MachineLearning \ env \ lib \ site-packages \ tensorflow_core \ python \ saved_model \ signature_serialization.py”，第195行 返回self._signatures [key] KeyError：“默认”

任何帮助将不胜感激。 谢谢

Answer 1

SavedModel与老式GraphDef的文件格式不同。他们不能那样交换。