如何从 .pb 文件为 opencv 中的 dnn 模块生成 .pbtxt 文件？

Posted 2023-02-23

【中文标题】如何从 .pb 文件为 opencv 中的 dnn 模块生成 .pbtxt 文件？

【英文标题】：How to generate .pbtxt file from a .pb file for dnn module in opencv?

【发布时间】：2019-10-26 07:54:21

【问题描述】：

我有一个自定义的 tensorflow 模型 .pb 文件。我想使用 dnn 模块将其导入 OpenCV。 dnn 模块函数需要 2 个参数：.pb 文件和.pbtxt 文件？如何从.pb 文件生成.pbtxt 文件？

cvNet = cv2.dnn.readNetFromTensorflow('FrozenMaster.pb', 'graph.pbtxt')

如何获得这个graph.pbtxt 文件？

【问题讨论】：

您从哪里获得 .pb 文件？如果您自己制作，您使用的是哪个版本的 TensorFlow？

【参考方案1】：

import cv2 
cv2.dnn.writeTextGraph('FrozenMaster.pb', 'graph.pbtxt')

此外，opencv github的以下链接也可能有用https://github.com/opencv/opencv/wiki/TensorFlow-Object-Detection-API