【问题标题】:No module named 'object_detection'没有名为“object_detection”的模块
【发布时间】: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 导入 o​​bject_detection,现在仅在 train.py 第 49 行出现错误

标签: python tensorflow object-detection object-detection-api


【解决方案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 命令。 希望这可以帮助你。如果您遇到任何问题,请告诉我。

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2021-12-17
    • 2018-06-30
    • 1970-01-01
    • 2018-08-14
    • 2019-03-28
    • 2017-12-30
    • 2018-01-06
    相关资源
    最近更新 更多