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

Question

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

要在opencv中运行，我要使用：

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

但是当我使用以下方式保存模型时：

model.save('./')

我获得了这些文件：

saved_model.pb | variables / variables.index | variables / variables.data-00000-of-00002 | variables / variables.data-00001-of-00002

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

你有什么想法吗？

谢谢！

Tanguy

Answer 1

  

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

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