对象检测：导出/导入推理时出错

埃里克森哦

我是机器学习的初学者，目前正在尝试遵循以下链接https://github.com/tensorflow/models/blob/master/object_detection/g3doc/exporting_models.md 中给出的教程

我已经完成了一个模型的训练，它给了我 model.ckpt 文件。下面是我在窗口命令提示符中输入的命令

{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

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

杰瑞B

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

trained_checkpoint_prefix = "{你的路径}\models\train\" \ --output_directory=output_inference_graph.pb \

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

尝试在“train\”之后添加更多的“\”-->“train\\”

{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 \

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。