Использование инструментов преобразования Tensorflow Graph

Когда я пытаюсь использовать инструмент преобразования графиков с FrozenGraph в Windows 10 через командную строку, я всегда получаю одну и ту же ошибку. Код:

C://tensorflow/bazel-bin/tensorflow/tools/graph_transforms/transform_graph ^
--in_graph=C://tensorflow-for-poets-2/tf_files/retrained_graph.pb ^
--out_graph=C://tensorflow-for-poets-2/new_optimized.pb ^
--inputs=Mul ^
--outputs=final_result ^
--transforms=' ^
   strip_unused_nodes ^
   fold_constants(ignore_errors=true) ^
   fold_batch_norms ^
   fold_old_batch_norms'

Ошибка:

2018-07-27 14:59:09.303184: E tensorflow/tools/graph_transforms/transform_graph.cc:217] Unknown argument strip_unused_nodes.
usage: C://tensorflow/bazel- bin/tensorflow/tools/graph_transforms/transform_graph

Как мне поступить? Мне нужно оптимизировать этот график для работы с tenorflowjs. Это InceptionV3, поэтому он довольно большой для мобильных устройств.

1 ответ

Решение

Неизвестный аргумент strip_unused_nodes

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

Другие вопросы по тегам