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