ImportError: cannot import name 'exporter_lib_v2'

Question

我在尝试从上次保存的检查点生成推理图时遇到问题。我运行:

python object_detection/exporter_main_v2.py --input_type image_tensor --pipeline_config_path C:/Train_Mask_RCNN/Tensorflow_API-Custom_Mask_RCNN/mask_rcnn_inception_v2_coco.config --trained_checkpoint_dir C:/Train_Mask_RCNN/Tensorflow_API-Custom_Mask_RCNN/CP/ --output_directory C:/Train_Mask_RCNN/Tensorflow_API-Custom_Mask_RCNN/IG

但它错误地说：

Traceback (most recent call last):
  File "exporter_main_v2.py", line 105, in <module>
    from object_detection import exporter_lib_v2
ImportError: cannot import name 'exporter_lib_v2'

工作目录是 models/research，exporter_lib_v2.py 就在预期的位置。这里似乎有什么问题？还有其他方法可以生成IG吗？谢谢。

Answer 1

要为正在寻找解决方案的人回答这个问题，您需要编辑 models/research/object_detection 文件夹中 exporter_main_v2.py 中的代码，位于

的第 105 行

from object_detection import exporter_lib_v2

至

import exporter_lib_v2

ImportError: cannot import name 'exporter_lib_v2'

ImportError: cannot import name 'exporter_lib_v2'

python

object-detection

tensorflow

tensorflow2.0