运行 export_inference_graph.py 抛出属性错误

Running export_inference_graph.py throws an attribute error

所以,在云端训练后,我尝试使用以下link导出推理图:

https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/exporting_models.md

我使用的代码是:

python export_inference_graph.py \--input_type image_tensor \--pipeline_config_path samples/configs/ssd_mobilenet_v1_pets \--trained_checkpoint_prefix training\model.ckpt-29809 \--output_directory crop_graph1

它抛出一个错误:

C:\Users\kannan\Desktop\Deep Learning Projects\GCP\models\research\object_detection>python export_inference_graph.py --input_type image_tensor --pipeline_config_path samples/configs/ssd_mobilenet_v1_pets --trained_checkpoint_prefix training\model.ckpt-29809 --output_directory crop_graph1 Traceback (most recent call last):
File "export_inference_graph.py", line 119, in tf.app.run() File "C:\Users\kannan\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\platform\app.py", line 41, in run flags_passthrough = f._parse_flags(args=args) File "C:\Users\kannan\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\platform\flags.py", line 45, in _parse_flags self._assert_all_required() File "C:\Users\kannan\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\platform\flags.py", line 78, in _assert_all_required self._assert_required(flag_name) File "C:\Users\kannan\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\platform\flags.py", line 74, in _assert_required raise AttributeError('Flag --%s must be specified.' % flag_name) AttributeError: Flag --output_directory must be specified.

知道如何解决这个问题吗?

我认为问题出在下面一行

\--trained_checkpoint_prefix training\model.ckpt-29809 \--output_directory crop_graph1

这个training\model.ckpt-29809应该有一个双引号,比如

"training\model.ckpt-29809"

我假设 python 正在处理 ckpt 和 29809 之间的额外“-” 或者删除反斜杠,因为它表示连续命令。

命令应在一行中,不带“\”

python export_inference_graph.py --input_type image_tensor --pipeline_config_path samples/configs/ssd_mobilenet_v1_pets --trained_checkpoint_prefix "training\model.ckpt-29809" --output_directory crop_graph1