如何从Tensorflow Objectdetection 2中的检查点导出冻结的\u推断\u graph.pb

Question

如何将经过训练的模型导出到冻结的_推断_graph.pb而不是保存的_model.pb，因为当我使用Tensorflow object detection v2附带的导出器_main _v2.py时，它会给我一个文件夹

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

在save_model中，我保存了_model.pb，但问题是我不能单独使用它进行推断，但我需要使用它附带的变量文件夹。这就是为什么我要问theire是否是一种将经过训练的模型导出到freezed_interference_graph.pb的方法，以便在不需要TF1中类似folder的变量的情况下使用它进行推理

Answer 1

我们在TensorFlow模型git中有引用，可以导出模型进行推理，并尝试使用它-https://github.com/tensorflow/models/blob/master/research/object_detection/export_inference_graph.py

Answer 2

不幸的是，TF2不支持export\u interference\u graph.py，它将执行您要求的操作，但只支持TF1。 exporter_main_v2.py不会导出任何冻结的图形

您可以做的是使用冻结图形

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

您还可以在计算机上的tensorflow\python\tools中找到文件freeze\u graph.py，该文件对如何使用它进行了很好的注释。 另外，您可以看看如何冻结它：freeze_graph_test 但是，TF2不再打算生成冻结图

编辑：可以找到很好的解决方法here