TensorfFlow对象检测API-ModuleNotFoundError：没有名为'object_detection'的模块

Question

在MACos上正确安装（遵循说明https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html），但运行脚本时我仍然不断收到此MNFE；

Traceback (most recent call last):
  File "generate_tfrecord.py", line 29, in <module>
    from object_detection.utils import dataset_util, label_map_util
ModuleNotFoundError: No module named 'object_detection'

还有其他人遇到同样的问题吗？在Windows上发现了很多东西，而不是在Mac上找到了。

Answer 1

在脚本的开头使用以下几行：

import sys
sys.path.extend([
r'path\to\models\research\object-detection',
r'path\to\models\research\tf_slim',
r'path\to\models\research',
r'path\to\models',
])

这些必须在此行之前 运行：

from object_detection.utils import dataset_util, label_map_util