0

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

正在加载 TF 图...

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

已加载图表。通过运行它从图中收集所有“常量”操作......完成。回溯（最近一次调用）：文件“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 行，在 _convert_pb_to_mlmodel raise ValueError("output名称：{}，已提供，但 Tensorflow 图不包含具有此名称的张量。".format(given_out_name)) ValueError：输出名称：softmax:0，已提供，但 Tensorflow 图不包含具有此名称的张量这个名字。

4

1 回答 1

1

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

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

于 2018-12-05T10:36:22.423 回答