错误：object_detection.utils ModuleNotFoundError

Question

我正在按照本教程（https://towardsdatascience.com/creating-your-own-object-detector-ad69dda69c85）创建自己的对象检测器。

在“为训练生成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. 转到文件夹“ research”。 打开终端： 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 /

export PYTHONPATH = $ PYTHONPATH：pwd：pwd / slim

Source Source2