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

Posted 2023-02-23

【中文标题】如何使用 tensorflow 2 生成 freeze_inference_graphe.pb 和 .pbtxt 文件

【英文标题】：How to generate frozen_inference_graphe.pb and .pbtxt files with tensorflow 2

【发布时间】：2020-03-05 21:11:39

【问题描述】：

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

bufferModel : 包含 pb 文件内容的缓冲区 (frozen_inference_graphe)

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

我发现的一切都依赖于“freeze_graph.py”或其他仅适用于 tensorflow 1.x 的解决方案。 tensorflow 2 应该怎么做？

【参考方案1】：

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