沒有名為“ object_detection”的模塊

Question

我下載了Tensorflow object_detection API。 我能夠運行該教程並查看結果。

但是，雖然我想訓練自己的數據，但是在此代碼處出現錯誤：

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

錯誤將如下所示：

追溯（最近一次呼叫最近）：來自object_detection.builders的文件“ train.py”，第49行，import data_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命令。 希望這可以幫助你。 如果您遇到任何問題，請告訴我。