对象检测:导出/导入用于推断时出错

时间:2017-10-24 01:39:58

标签: python tensorflow object-detection

我是机器学习的初学者,目前正在尝试按照以下链接中提供的教程https://github.com/tensorflow/models/blob/master/object_detection/g3doc/exporting_models.md

我已经完成了一个模型的训练,该模型给了我model.ckpt files.below是我在窗口命令提示符下输入的命令

{Your path}\tensor\flow\models\research>
python object_detection\export_inference_graph.py\
--input_type=image_tensor \
--pipeline_config_path="{Your path}\model\ssd_mobilenet_v1_pets.config" \
--trained_checkpoint_prefix="{Your path}\models\train\" \
--output_directory=output_inference_graph.pb \
1>mloutput.txt 2>mlerror.txt

所以当我检查错误文件时,这就是我发现的:

Traceback (most recent call last):
  File "object_detection\export_inference_graph.py", line 106, in <module>
    tf.app.run()
  File "C:\Users\ericsen\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\platform\app.py", line 48, in run
    _sys.exit(main(_sys.argv[:1] + flags_passthrough))
  File "object_detection\export_inference_graph.py", line 95, in main
    assert FLAGS.output_directory, '`output_directory` is missing'
AssertionError: `output_directory` is missing

我真的不明白为什么缺少输出目录。这纯粹是我的错误还是一个错误。我们将非常感谢您的反馈和帮助。谢谢

1 个答案:

答案 0 :(得分:0)

我已尝试打印所有参数并查看:

  

trained_checkpoint_prefix =“{您的路径} \ models \ train \”\ --output_directory = output_inference_graph.pb \

它似乎在“{Your path} \ models \ train \”

中缺少“\”

尝试在“train \”之后添加更多“\” - &gt; “列车\\”

{Your path}\tensor\flow\models\research>
python object_detection\export_inference_graph.py\
--input_type=image_tensor \
--pipeline_config_path="{Your path}\model\ssd_mobilenet_v1_pets.config" \
--trained_checkpoint_prefix="{Your path}\models\train\\" \ 
--output_directory=output_inference_graph.pb \