在colaboratory训练张量流检测api

时间:2018-05-23 21:21:11

标签: python tensorflow google-colaboratory

我试图在colaboratory训练张量流物体检测api。

当我运行train.py时,运行train.py ...'我收到错误'重复标记错误:标记' master'定义了两次。

当我运行train.py作为'!python train.py ...'我收到错误'没有命名部署的模块'。

我已使用以下方式设置路径: 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脚本。

1 个答案:

答案 0 :(得分:1)

下面的代码片段用于在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

`

在google colab上,需要执行上面的一系列行,以便让你的对象检测api启动并运行。 安装的实际详细信息位于here

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

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

从官方文档中,所有这些脚本都在

中运行
  

/模型/研究

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