【发布时间】:2019-07-31 16:48:42
【问题描述】:
我下载了 Tensorflow object_detection API。我能够运行教程并查看结果。
但是,虽然我想训练自己的数据,但在此代码中出现错误:
python3 train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_pets.config
报错如下:
Traceback(最近一次调用最后一次):文件“train.py”,第 49 行,在 from object_detection.builders import dataset_builder ModuleNotFoundError: No module named 'object_detection'
这里是 train.py 中的代码 sn-p:
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
【问题讨论】:
-
你可以试试
pip freeze看看你能不能看到object_detection吗? -
您是否在代码顶部导入了 object_detection api
-
运行 pip freeze 后,它的 show object-detection==0.1
-
发布代码sn-p。还要确保您已正确导入所有文件。
-
我从 object_detection import trainer 导入 object_detection,现在仅在 train.py 第 49 行出现错误
标签: python tensorflow object-detection object-detection-api