Я пытаюсь преобразовать файл.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

График загружен. Собрать все операции 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", выходной_переменный__интакс. имя: {}, было предоставлено, но граф Tensorflow не содержит тензор с этим именем.".format(данное_имя)) ValueError: имя выхода: softmax:0, было предоставлено, но граф Tensorflow не содержит тензор с это имя.

Answer 1

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

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