如何使用tensorflow 2生成frozen_inference_graphe.pb和.pbtxt文件

Question

我想在opencv中使用我自己的tensorflow 2 / keras模型（cv.dnn.readNetFromTensorflow（bufferModel [，bufferConfig]）。但是，我没有设法生成所需的文件：

bufferModel：包含pb文件（frozen_inference_graphe）内容的缓冲区

bufferConfig：包含pbtxt文件（模型配置文件）内容的缓冲区

我发现的所有内容都依赖于“ freeze_graph.py”或仅适用于tensorflow 1.x的其他解决方案。我该如何使用tensorflow 2？

Answer 1

使用.pb时将生成keras.callbacks.ModelCheckpoint()文件。 但是，我不知道如何创建.pbtxt文件。