python - 错误 : 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. 转到文件夹“研究”。打开终端:cd models-master/research

3. (对于 3.6python)python3.6 setup.py build

   (对于 3.6python)sudo python3.6 setup.py install

4. 您将看到:“已完成对象检测的处理依赖项==0.1”

然后执行以下操作

Add Libraries to PYTHONPATH

When running locally, the tensorflow/models/research/ and slim directories should be appended to PYTHONPATH. This can be done by running the following from tensorflow/models/research/:

来自 tensorflow /模型/研究/

export PYTHONPATH=$PYTHONPATH:pwd:pwd/slim

Source Source2