错误：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