首页 文章

在colaboratory训练张量流检测api

提问于
浏览 次
1

我正在尝试在colaboratory上训练tensorflow物体检测api .

当我运行train.py作为'％run train.py ...'时，我收到错误'重复标志错误：标志'主'被定义两次 .

当我将train.py作为'！python train.py ...'运行时，我收到错误'no module named deployment' .

我已使用：import sys设置路径

sys.path.append('/content/models/research/slim')
sys.path.append('/content/models/research')
sys.path.append('/content/models/research/object_detection')

另外我从object_detection目录运行train . 我不确定还需要做什么才能成功运行train.py脚本 .

python tensorflow google-colaboratory

1 回答

  • 0

    下面的代码段用于在colab上专门设置tensorflow对象检测api .

    `

    !apt-get install protobuf-compiler python-pil python-lxml python-tk
!pip install Cython
!pip install jupyter
!pip install matplotlib

!git clone https://github.com/tensorflow/models.git

%cd /content/models/research

!protoc object_detection/protos/*.proto --python_out=.

%set_env PYTHONPATH=/content/models/research:/content/models/research/slim

!python object_detection/builders/model_builder_test.py

    `

    在谷歌colab上，需要执行上面的一系列行，以便让你的对象检测api启动并运行 . 安装的实际细节是here

    以下行指定如何运行火车脚本

    python object_detection/train.py \
    --logtostderr \
    --pipeline_config_path=${PATH_TO_YOUR_PIPELINE_CONFIG} \
    --train_dir=${PATH_TO_TRAIN_DIR}

    从官方文档中，所有这些脚本都是从内部运行的

    / models / research

    目录 . 不是来自对象检测文件夹本身 .

    回复于

相关问题