1. 程式人生 > >ubuntu16.04 'No module named 'object_detection''解決方案

ubuntu16.04 'No module named 'object_detection''解決方案

博主在使用tensorflow的Tensorflow object detection API 訓練自己的資料集時,出現一下error:

Traceback (most recent call last):
  File "train.py", line 49, in <module>
    from object_detection.builders import dataset_builder
ModuleNotFoundError: No module named 'object_detection'

博主在網上找了很久,都沒找到相關ubuntu環境下的解決方案,於是參考了windows的解決方案。

即在anaconda安裝目錄下的site-packages資料夾下建立一個tensorflow_model.pth檔案,檔案內容寫models的路徑,已經slim的路徑,博主的路徑是:

/home/ai-boy/anaconda3/lib/python3.6/site-packages/tensorflow/models/research
/home/ai-boy/anaconda3/lib/python3.6/site-packages/tensorflow/models/research/slim

新增完後完美解決。