Warning: file_get_contents(/data/phpspider/zhask/data//catemap/9/opencv/3.json): failed to open stream: No such file or directory in /data/phpspider/zhask/libs/function.php on line 167

Warning: Invalid argument supplied for foreach() in /data/phpspider/zhask/libs/tag.function.php on line 1116

Notice: Undefined index: in /data/phpspider/zhask/libs/function.php on line 180

Warning: array_chunk() expects parameter 1 to be array, null given in /data/phpspider/zhask/libs/function.php on line 181
保存tensorflow 2.0模型并在opencv 4中使用_Opencv_Tensorflow - Fatal编程技术网

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

保存tensorflow 2.0模型并在opencv 4中使用,opencv,tensorflow,Opencv,Tensorflow,我目前使用tensorflow 2.0编写模型,并希望使用opencv 4运行它们(我希望比较性能)。但是我找不到一种方法将我的tensorflow模型转换为opencv 对于在opencv中运行,我要使用: cv2.dnn.readNetFromTensorflow('saved_model.pb','saved_model.pbtxt') 但当我使用以下工具保存模型时: model.save('./') 我获得以下文件: 已保存的_model.pb |变量/变量.index |变量/变量.d

我目前使用tensorflow 2.0编写模型,并希望使用opencv 4运行它们(我希望比较性能)。但是我找不到一种方法将我的tensorflow模型转换为opencv

对于在opencv中运行,我要使用:

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

但当我使用以下工具保存模型时:

model.save('./')

我获得以下文件:

已保存的_model.pb |变量/变量.index |变量/变量.data-00000-of-00002 |变量/变量.data-00001-of-00002

我有一个my.pb,但没有my.pbtxt。如何写入此文件?根据opencv文档,该文件是文本图形定义。我已经试着用
model.to_json()
但它不起作用:/

你有什么想法吗

提前谢谢

坦古

此外,OpenCV需要一个基于 .pb,即.pbtxt。可以导入您自己的模型和 使用以下文件之一生成您自己的.pbtxt文件 从OpenCV Github存储库

以下是教程的链接: 我自己也没试过,但似乎合法。
例如,
tf\u text\u graph\u ssd.py
完成了工作。

您的链接仅适用于对象检测api tensorflow模型,下面是另一个相同的教程:我使用了此方法,它可以工作,但仅适用于选定的对象检测模型。转换说明和工作模型在我共享的链接中。OP正在创建自定义模型,我还没有看到opencv与自定义模型一起使用。@Shortipher3,你是对的,但我认为主要问题是缺少
.pbtxt
。谢谢你的回答,但我的问题是缺少.pbtxt:/@TanguyHardelin,那么,你知道如何使用我发送的链接中的脚本生成它了吗?是的,我尝试过,但你的脚本似乎只适用于ssd/rcnn。我想用RESNet模型进行测试(比较opencv是否做了一些好的优化)。是否可以为转换所有类型的模型编写通用脚本?