没有名为“object_detection”的模块

Question

我下载了Tensorflow object_detection API。我能够运行这个教程并看到结果。

但是，当我想训练自己的数据时，这里的代码出现了一个错误：

python3 train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_pets.config

该错误将如下所示：

回溯(最近一次调用)：文件"train.py"，第49行，从object_detection.builders导入dataset_builder ModuleNotFoundError:没有名为'object_detection‘的模块

在这里，来自train.py的代码片段：

import functools 
import json
import os
import tensorflow as tf 

from object_detection.builders import dataset_builder
from object_detection.builders import graph_rewriter_builder
from object_detection.builders import model_builder
from object_detection.legacy import trainer
from object_detection.utils import config_util

信息：

我正在使用Tensorflow 1.10和Windows 10

Note

我运行这段代码，但是它对我不起作用。

设置PYTHONPATH=$PYTHONPATH:pwdpwd/slim

Answer 1

您可以尝试以下步骤。切换到对象检测目录，激活virtualenv，然后执行以下操作

export PYTHONPATH=$PYTHONPATH:home/<username>/<path>/models/research
export PYTHONPATH=$PYTHONPATH:home/<username>/<path>/models
export PYTHONPATH=$PYTHONPATH:home/<username>/<path>/research/slim
PATH=$PATH:$PYTHONPATH

cd .. (Make sure you are now in the research directory)
python setup.py build
python setup.py install

现在切换到object_detection目录，再试一次train.py命令。希望这能帮到你。如果你遇到任何问题，请告诉我。