Я пытаюсь преобразовать файл .pb в файл .mlmodel.Я получаю сообщение об ошибке "Граф Tensorflow не содержит тензор с этим именем"

Question

Я попытался использовать tfcoreml, и ошибка:

Загрузка графика TF ...

2018-12-05 11: 16: 50.591360: I tenorflow / core / platform / cpu_feature_guard.cc: 141] Ваш ЦП поддерживает инструкции, которые этот двоичный файл TensorFlow не был скомпилирован для использования: AVX2 FMA

Graph Loaded.Собрать все операции Const из графика, запустив его .... Готово.Traceback (последний вызов был последним): файл "converter.py", строка 4, в output_feature_names = ['softmax: 0']) файл "/ root / tenorflow-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 ", выходной параметр_2_rever_rever_rever_rever_re_t_t_t_t_t_t_t_t_t_t_t_t_t_t_t_t_t_t_rt_t_rt_t_t_t_t_rt_t_t_t_rt_rt_2имя: {}, было предоставлено, но граф Tensorflow не содержит тензор с этим именем. ". format (данное_имя)) ValueError: имя выхода: softmax: 0, было предоставлено, но граф Tensorflow не содержит тензор сэто имя.

Answer 1

Когда вы звоните tfcoreml.convert(), вам нужно указать имя тензора с выводом модели.Вы поставили "softmax:0", вероятно, потому что вы видели это в учебнике где-то.Но, видимо, это не имя вывода вашего собственного графика TensorFlow.

Вы можете использовать инструмент под названием Netron , чтобы проверить ваш график, чтобы увидеть, как называется выходной тензор.