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

注意

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

设置PYTHONPATH = $ PYTHONPATH：pwd：pwd / 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命令。希望这可以帮助你。如果您遇到任何问题，请告诉我。

没有名为“ object_detection”的模块

1 个答案: