转换 TensorFlow* 模型

Question

我想将我的 1 个 tensorflow 模型转换为 IR，目前我正在按照此处的说明进行操作： https://docs.openvinotoolkit.org/latest/_docs_MO_DG_prepare_model_convert_model_Convert_Model_From_TensorFlow.html

我使用的模型是元图和ubuntu 16.04

我跑的线偏了： python3 mo_tf.py --input_meta_graph .meta

然后会报错： [ERROR] 运行替换器“None”（）期间发生异常：AssignSub 节点模型_0/resnet_v1_50/block4/unit_1/bottleneck_v1/shortcut/BatchNorm/AssignMovingAvg 出来的数据流边缘

你们能帮帮我吗？谢谢大家

Answer 1

您是否在转换前冻结了模型？请查看如何使用来自https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/freeze_graph.py 的说明冻结您的模型并重试。