我训练了一个用于图像分类的 resnet 模型,并希望冻结图形。当我使用 freeze_graph 工具时,它给了我以下错误:
Traceback (most recent call last):
File "/media/goodvirus/34e9f1a6-6a30-4087-b582-b02e43a9aa82/DeepLearning/Entwicklungen/venv/bin/freeze_graph", line 11, in <module>
sys.exit(run_main())
File "/media/goodvirus/34e9f1a6-6a30-4087-b582-b02e43a9aa82/DeepLearning/Entwicklungen/venv/local/lib/python2.7/site-packages/tensorflow/python/tools/freeze_graph.py", line 408, in run_main
app.run(main=my_main, argv=[sys.argv[0]] + unparsed)
File "/media/goodvirus/34e9f1a6-6a30-4087-b582-b02e43a9aa82/DeepLearning/Entwicklungen/venv/local/lib/python2.7/site-packages/tensorflow/python/platform/app.py", line 125, in run
_sys.exit(main(argv))
File "/media/goodvirus/34e9f1a6-6a30-4087-b582-b02e43a9aa82/DeepLearning/Entwicklungen/venv/local/lib/python2.7/site-packages/tensorflow/python/tools/freeze_graph.py", line 407, in <lambda>
my_main = lambda unused_args: main(unused_args, flags)
File "/media/goodvirus/34e9f1a6-6a30-4087-b582-b02e43a9aa82/DeepLearning/Entwicklungen/venv/local/lib/python2.7/site-packages/tensorflow/python/tools/freeze_graph.py", line 301, in main
flags.saved_model_tags, checkpoint_version)
File "/media/goodvirus/34e9f1a6-6a30-4087-b582-b02e43a9aa82/DeepLearning/Entwicklungen/venv/local/lib/python2.7/site-packages/tensorflow/python/tools/freeze_graph.py", line 283, in freeze_graph
checkpoint_version=checkpoint_version)
File "/media/goodvirus/34e9f1a6-6a30-4087-b582-b02e43a9aa82/DeepLearning/Entwicklungen/venv/local/lib/python2.7/site-packages/tensorflow/python/tools/freeze_graph.py", line 145, in freeze_graph_with_def_protos
var_list=var_list, write_version=checkpoint_version)
File "/media/goodvirus/34e9f1a6-6a30-4087-b582-b02e43a9aa82/DeepLearning/Entwicklungen/venv/local/lib/python2.7/site-packages/tensorflow/python/training/saver.py", line 1078, in __init__
self.build()
File "/media/goodvirus/34e9f1a6-6a30-4087-b582-b02e43a9aa82/DeepLearning/Entwicklungen/venv/local/lib/python2.7/site-packages/tensorflow/python/training/saver.py", line 1090, …Run Code Online (Sandbox Code Playgroud)