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.

Еще мне очень хотелось бы понять, почему у меня не работает командная строка:(

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