我正在尝试在colaboratory上训练Tensorflow对象检测API。

当我将train.py运行为'%run train.py ...'时,收到错误消息'Duplicate flag error:标志'master'被定义了两次。

当我将train.py运行为'!python train.py ...'时,收到错误消息“没有名为部署的模块”。

我使用以下方法设置了路径:
导入系统

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


另外,我正在从object_detection目录运行火车。我不确定要成功运行train.py脚本还需要做什么。

最佳答案

以下代码段专门用于在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}


根据官方文档,所有这些脚本均在


  /型号/研究


目录。并非来自对象检测文件夹本身。

关于python - 在colaboratory上训练Tensorflow检测API,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/50497648/

10-12 21:53
查看更多