Import Error: cannot import name 'model_lib_v2' from 'object_detection'

Question

我正在 google colab 中研究对象检测模型，并且我遵循 here 中列出的大部分说明。

为了训练模型，我尝试使用：

!python model_main_tf2.py 
!python --model_dir=models/SSD_640
!python --pipeline_config_path=models/SSD_640/pipeline.config

但是，我收到以下错误：

/content/drive/MyDrive/Workspace
2021-06-28 11:33:33.510377: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library libcudart.so.11.0
Traceback (most recent call last):
  File "model_main_tf2.py", line 32, in <module>
    from object_detection import model_lib_v2
ImportError: cannot import name 'model_lib_v2' from 'object_detection' (/usr/local/lib/python3.7/dist-packages/object_detection/__init__.py)

提到的文件 model_lib_v2.py 在以下路径中可用 - Workspace/models/research/model_lib_v2.py

我尝试使用以下代码调整 cd 以指向工作区文件夹：

%cd '/content/drive/MyDrive/Workspace'

我也一直试图通过再次安装 object_detection 来修复它，但这并没有帮助。我也确保路径是正确的，而且似乎没有任何不一致。

如有任何帮助，我们将不胜感激！提前致谢。

Answer 1

model_lib_v2.py 必须在文件夹 object_detection.

中

尝试添加到 PYTHONPATH。

您可以从这里获取文件。

https://github.com/tensorflow/models/tree/master/research/object_detection

Import Error: cannot import name 'model_lib_v2' from 'object_detection'

Import Error: cannot import name 'model_lib_v2' from 'object_detection'

python

object-detection

tensorflow

google-colaboratory

tensorflow2.0