如何从 Tensorflow Objectdetection 2 中的检查点导出 frozen_inference_graph.pb

Question

我如何将经过训练的 model 导出到 frozen_inference_graph.pb 而不是 saved_model.pb，因为当我使用 Tensorflow object detection v2 附带的 exporter_main_v2.py 时，它会给我一个文件夹

├─ exported-models/
   └─ my_model/ 
      ├─ checkpoint/
      ├─ saved_model/
            └─ assets/
            ├─ variables/
            └─ saved_model.pb
      └─ pipeline.config

在 save_model 中我有 saved_model.pb 但问题是我不能单独使用它进行推理但我需要使用它附带的变量文件夹。 这就是为什么我要问他们是否有办法将经过训练的 model 导出到 frozen_inference_graph.pb 以使用它进行推理，而无需 TF1 中的变量文件夹。

Answer 1

不幸的是，TF2 不支持 export_inference_graph.py，它可以完全满足您的要求 - 但仅适用于 TF1。 exporter_main_v2.py 不会导出任何冻结图。

你可以做的是使用冻结你的图表

from tensorflow.python.tools import freeze_graph
freeze_graph.freeze_graph(...)

您还可以在您的计算机上的 tensorflow\python\tools 中找到文件 freeze_graph.py，其中对如何使用它有很好的评论。 此外，您还可以查看如何冻结它： freeze_graph_test但是，TF2 不再打算生成冻结图！

编辑：可以在这里找到很好的解决方法

Answer 2

我们在 TensorFlow model git 中有参考导出 model 用于推理尝试使用它- https://github.com/tensorflow/models/blob/master/research/object_detection/export_inference_graph.py