我正在尝试将 .pb 文件转换为 .mlmodel 文件。我收到错误“Tensorflow 图不包含具有此名称的张量”

取消网格

我尝试使用 tfcoreml 并且错误是:

正在加载 TF 图...

2018-12-05 11:16:50.591360: I tensorflow/core/platform/cpu_feature_guard.cc:141] 您的 CPU 支持该 TensorFlow 二进制文件未编译使用的指令:AVX2 FMA

图表加载。通过运行它从图中收集所有“Const”操作......完成。回溯(最近一次调用):文件“converter.py”,第 4 行,在 output_feature_names = ['softmax:0'])文件“/root/tensorflow-for-poets-2/tf_files/tf-coreml/tfcoreml/ _tf_coreml_converter.py”,第 586 行,在转换 custom_conversion_functions=custom_conversion_functions 中)文件“/root/tensorflow-for-poets-2/tf_files/tf-coreml/tfcoreml/_tf_coreml_converter.py”,第 272 行,在 raise_toconvert_model 中,值名称:{},已提供,但 Tensorflow 图不包含具有此名称的张量。".format(given_out_name)) ValueError: 输出名称:softmax:0,已提供,但 Tensorflow 图不包含具有此名称的张量这个名字。

马蒂斯·霍勒曼斯

当您调用时,tfcoreml.convert()您需要提供带有模型输出的张量名称。您提供了"softmax:0",可能是因为您在某处的教程中看到了它。但显然这不是您自己的 TensorFlow 图的输出名称。

您可以使用名为Netron的工具来检查您的图形以查看输出张量的名称。

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章

Related 相关文章

热门标签

归档