保存 tensorflow 2.0 模型并在 opencv 4 中使用它们

Question

我目前使用 tensorflow 2.0 编写我的模型，我想用 opencv 4 运行它们（我想比较性能）。但是我找不到将我的 tensorflow 模型转换为 opencv 的方法。

为了在 opencv 中运行，我想使用：

cv2.dnn.readNetFromTensorflow('saved_model.pb', 'saved_model.pbtxt')

但是当我使用以下命令保存模型时：

model.save('./')

我得到这个文件：

save_model.pb | 变量/variables.index | 变量/variables.data-00000-of-00002 | 变量/variables.data-00001-of-00002

我有我的 .pb 但没有我的 .pbtxt。怎么可能写这个文件？根据 opencv 文档，这个文件是文本图定义。我已经尝试用 .pbtxt 写一个， model.to_json() 但没有用：/

你有什么想法 ？

提前致谢 ！

唐基

Answer 1

此外，OpenCV 需要一个基于 .pb、.pbtxt 的额外配置文件。可以使用 OpenCV Github 存储库中的以下文件之一导入您自己的模型并生成您自己的 .pbtxt 文件...

这是教程的链接： https: //jeanvitor.com/tensorflow-object-detecion-opencv/ 我自己没有尝试过，但似乎是合法的。例如，tf_text_graph_ssd.py工作是否完成。