保存tensorflow 2.0模型并在opencv 4中使用
我目前使用tensorflow 2.0编写模型,并希望使用opencv 4运行它们(我希望比较性能)。但是我找不到一种方法将我的tensorflow模型转换为opencv 对于在opencv中运行,我要使用:保存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
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是否做了一些好的优化)。是否可以为转换所有类型的模型编写通用脚本?