錯誤：object_detection.utils ModuleNotFoundError

Question

我正在按照本教程（ https://towardsdatascience.com/creating-your-own-object-detector-ad69dda69c85 ）創建我自己的 object 檢測器。

關於“為訓練生成 TFRecords”主題，當我嘗試運行此代碼時：

python generate_tfrecord.py --csv_input=images\train_labels.csv --image_dir=images\train --output_path=train.record

我收到了這個錯誤：

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

我的文件夾結構：

-generate_tfrecord.py
-images/
--test_labels.csv
--train_labels.csv
--test/
---*files*
--train/
---*files*

Answer 1

檢查這個：

1. git 克隆

   https://github.com/tensorflow/models

2. Go 到文件夾“研究”。 打開終端： cd models-master/research

3. （對於 3.6python） python3.6 setup.py build

   （對於 3.6python） sudo python3.6 setup.py install

4. 您將看到：“對象檢測的完成處理依賴項==0.1 ”

然后進行以下操作

將庫添加到 PYTHONPATH

在本地運行時，應將 tensorflow/models/research/ 和 slim 目錄附加到 PYTHONPATH。 這可以通過從 tensorflow/models/research/ 運行以下命令來完成：

來自 tensorflow/models/research/

導出 PYTHONPATH=$PYTHONPATH: pwd : pwd /slim

來源 來源2