TensorFlow freeze_graph: обязательный позиционный аргумент 'unused_args'
Я использую TensorFlow 1.7 с Python 3.6.5 на Mac с High Sierra.
Я обучил свою первую модель MNIST, поэтому я в основном
- файл graph.pbtxt со структурой графа CNN
- некоторые файлы model.ckpt-21000 (.meta, .index .data)
Я попытался заморозить график, используя команду командной строки freeze_graph на моем bash:
freeze_graph
--input_graph=/…/graph.pbtxt
--input_checkpoint=/…/model.ckpt-21000
--input_binary=false
--output_graph=/…/frozen_mnist.pb
--output_node_names=softmax_tensor
Но я получил эту ошибку:
Traceback (most recent call last):
File “/usr/local/bin/freeze_graph”, line 11, in <module>
sys.exit(main())
TypeError: main() missing 1 required positional argument: ‘unused_args’
Я не совсем уверен, чего мне там не хватает. Я совершенно уверен, что я использую правильный синтаксис.
1 ответ
Решение
Я нашел обходной путь, чтобы заморозить мой график. Я публикую его здесь, так что если кто-то сталкивается с той же проблемой, он может использовать это.
Вместо
freeze_graph
--input_graph=/…/graph.pbtxt
--input_checkpoint=/…/model.ckpt-21000
--input_binary=false
--output_graph=/…/frozen_mnist.pb
--output_node_names=softmax_tensor
использование
python3 -m tensorflow.python.tools.freeze_graph
--input_graph=/…/graph.pbtxt
--input_checkpoint=/…/model.ckpt-21000
--input_binary=false
--output_graph=/…/frozen_mnist.pb
--output_node_names=softmax_tensor
Так что в основном вместо команды freeze_graph я просто использовал python3 -m tenorflow.python.tools.freeze_graph.
Еще мне очень хотелось бы понять, почему у меня не работает командная строка:(