错误: 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. 转到文件夹“研究”。开放式终端:光盘模型-硕士/研究
3. (用于3.6python) python3.6 setup.py构建 (用于3.6python) sudopython3.6 setup.py安装
4. 您将看到：“完成对象检测=0.1的处理依赖项”

然后进行折叠

向PYTHONPATH添加库 在本地运行时，tensorflow/models/research/和slim目录应该附加到PYTHONPATH。这可以通过从tensorflow/models/research/运行以下步骤实现：

来自tensorflow/模型/研究/

导出PYTHONPATH=$PYTHONPATH:pwdpwd/slim

来源 Source2