实验室安装Tensorflow Object Detection Api

时间:2018-02-07 11:55:55

标签: tensorflow object-detection google-colaboratory

我在Google Colaboratory中成功执行了Tensorflow中的培训模型和图像识别笔记本。 现在我想用Object Detection Api开始一个新笔记本。当我执行我的代码时,我得到以下错误:

ModuleNotFoundError: No module named 'object_detection'

如何在Colaboratory中安装Object Detection Api?我按照安装instructions但我无法执行:

# From tensorflow/models/research/
protoc object_detection/protos/*.proto --python_out=.

3 个答案:

答案 0 :(得分:5)

以下示例笔记本显示了TensorFlow对象检测API的安装和配置:

https://colab.research.google.com/drive/1kHEQK2uk35xXZ_bzMUgLkoysJIWwznYr

退出网站上的安装说明包括直接修改sys.path并使用model_builder_test.py执行%run。造成这些差异的原因是,在Colab中运行时,您已经使用了Python解释器,因此您不必担心为将来的python shell调用修改环境。 / p>

答案 1 :(得分:3)

您忘了添加路径 苗条文件夹

如果您运行本地 https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md

export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim

Colab

import sys
sys.path.append('/content/base_folder/slim')

注意:Google Colaboratory还需要先安装一些必要的软件包:

!apt-get install -y -qq protobuf-compiler python-pil python-lxml

我的例子:https://colab.research.google.com/drive/1EFtTACXnWUoaGGAVqCwYS_JS-6Jr6upg#scrollTo=Z2GjW06y_6gO

答案 2 :(得分:0)

试试这个:

!pip install tensorflow-object-detection-api