opencv - 如何使用 Tensorflow 2 使用 graph_transform？

Question

我使用 Tensorflow 2.2.0 和 SSD MobileNet V2 FPNLite 640x640 训练了一个对象检测模型。它在我的 conda 环境中工作。但我想通过使用 readNetFromTensorflow(---.pb,---.pbtxt) 将此模型与 openCV dnn 一起使用。所以我将我的 saved_model.pb 转换为 frozen_graph.pb 。现在我必须将此 .pb 文件转换为 ---.pbttxt 文件以与 opencv dnn 一起使用。有一个名为 tf_text_graph_ssd.py 的脚本。通过使用此脚本，我可以使用 freeze_graph.pb 创建 ---.pbtxt 。但我的问题是这个 tf_text_graph_ssd 只与 tensorflow1.x 一起使用，因为一个名为 graph_transform 的库。所以我因此而出错。我尝试了很多方法来解决这个问题。有人能帮助我吗。

这里的错误：

错误照片

tf_text_graph_ssd：

脚本照片

我在用：

Cuda 10.1 cuDnn 7.6.5 ubuntu 20.04 tensorflow 2.2.0 opencv 4.2.0 gpu Nvidia GTX1650

我不能使用 tensorflow 1，因为我的 gpu 卡必须使用 cuda 10.1。如果我使用 tensorflow 1，我不能使用我的 gpu，因为 tensorflow 1.15 支持 cuda 10.0 。

标签： opencvtensorflow2.0object-detection