运行 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
所以,在云端训练后,我尝试使用以下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