我必须使用带有 OpenCV 框架的 Tensorflow 2.X 模型(带有 C++ 的 v.4.X)。
为此,我需要一个.pb文件或一个.pb和一个.pbtxt文件,而不是像我拥有的那样的 Tensorflow 保存模型。
所以我的问题是:有没有办法将保存的模型转换为 OpenCV 可以读取的格式?比如,也许,一个caffe模型?
我尝试使用MMdnn但它给了我一个奇怪的错误:
Traceback (most recent call last):
File "/usr/local/bin/mmconvert", line 8, in <module>
sys.exit(_main())
File "/usr/local/lib/python3.5/dist-packages/mmdnn/conversion/_script/convert.py", line 102, in _main
ret = convertToIR._convert(ir_args)
File "/usr/local/lib/python3.5/dist-packages/mmdnn/conversion/_script/convertToIR.py", line 62, in _convert
from mmdnn.conversion.tensorflow.tensorflow_parser import TensorflowParser
File "/usr/local/lib/python3.5/dist-packages/mmdnn/conversion/tensorflow/tensorflow_parser.py", line 15, in <module>
from tensorflow.tools.graph_transforms import TransformGraph
ImportError: No module named 'tensorflow.tools.graph_transforms'
Run Code Online (Sandbox Code Playgroud)
我想这是因为它是用 Tensorflow 1.X 开发和测试的。
编辑:我也有相对的 Keras 模型(现在它与 Tensorflow …