【发布时间】:2019-09-20 19:04:55
【问题描述】:
我正在从事一个检测眼睛区域的项目,并且我有一个冻结的推理模型。 我正在尝试将此模型转换为 .mlmodel(使用 tfcoreml),但出现此错误:
ValueError: output name: logits/BiasAdd 已提供,但 Tensorflow 图不包含具有此名称的张量。
+ 这是输入张量(input_to_float):
+ 这是输出张量(logits/BiasAdd):
这是我用于转换的代码:
import tfcoreml as tf_converter
tf_converter.convert(tf_model_path='C:\\Users\\User\\cnn-facial-landmark\\irismodel\\freeze2\\frozen_inf_model_iris.pb',
mlmodel_path='irismodelios.mlmodel',
output_feature_names=['logits/BiasAdd'],
input_name_shape_dict={'input_to_float': [1, 112, 112, 3]})
我该如何解决这个错误?
【问题讨论】:
-
试试
logits/BiasAdd:0,最后加上:0。 -
当我尝试
logits/BiasAdd:0时,我得到其他错误 >NotImplementedError: Unsupported Ops of type: OneShotterator,IteratorGetNext,Shape,Pack
标签: python tensorflow model coreml