【问题标题】:Error running export_inference_graph.py in Object Detection API在对象检测 API 中运行 export_inference_graph.py 时出错
【发布时间】:2018-01-11 08:50:25
【问题描述】:

当我使用object_detection 文件夹或models/research 文件夹中的正确路径运行export_inference_graph.py 时,如下所示:

python export_inference_graph \    # object_detection/ (if from /research)
    --input_type image_tensor \
    --pipeline_config_path path/to/ssd_inception_v2.config \
    --trained_checkpoint_prefix path/to/model.ckpt \
    --output_directory path/to/exported_model_directory

然后我得到以下错误:

  File "object_detection/export_inference_graph.py", line 71, in <module>
    from object_detection import exporter
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/models/research/object_detection/exporter.py", line 28, in <module>
    from object_detection.builders import model_builder
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/models/research/object_detection/builders/model_builder.py", line 17, in <module>
    from object_detection.builders import anchor_generator_builder
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/models/research/object_detection/builders/anchor_generator_builder.py", line 20, in <module>
    from object_detection.protos import anchor_generator_pb2
ImportError: cannot import name anchor_generator_pb2

anchor_generator_builder.py 内部,它尝试导入anchor_generator_pb2,但不存在同名文件。

附加信息: - 是的,我添加了以下库:

 # From tensorflow/models/research/
export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim

- 是的,我提取了最新的tensorflow/models 版本

请告诉我如何正确导出新的 Frozen Graph,我认为这应该很容易,但它会停止我的整个工作。

提前谢谢大家

【问题讨论】:

    标签: python tensorflow object-detection-api


    【解决方案1】:

    好的,案件结案。

    我忘了编译 Protobuf。

    # From tensorflow/models/research/
    protoc object_detection/protos/*.proto --python_out=.
    

    对不起,我的错

    【讨论】:

      猜你喜欢
      • 2019-04-16
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多