小编Doc*_*h88的帖子

在 OpenCV 上使用 Tensorflow 2.X 模型

我必须使用带有 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 …

c++ opencv tensorflow

7
推荐指数
1
解决办法
2591
查看次数

标签 统计

c++ ×1

opencv ×1

tensorflow ×1