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