tensorflow - Colaboratory 安装 Tensorflow 对象检测 Api

Question

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

ModuleNotFoundError: No module named 'object_detection'

如何在 Colaboratory 中安装对象检测 API？我按照安装说明进行操作，但无法执行：

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

Answer 1

这是一个示例笔记本，显示了 TensorFlow 对象检测 API 的安装和配置：

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

与网站上安装说明的不同之处包括sys.path直接修改和model_builder_test.py使用%run. 造成这些差异的原因是，在 Colab 中运行时，您已经处于 Python 解释器中，因此您无需担心为将来的 shell 调用而修改环境python。

Answer 2

你只是忘了添加路径苗条文件夹

如果您在本地运行 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

Answer 3

尝试这个 ：

!pip install tensorflow-object-detection-api

Answer 4

运行：

• 步骤1：

import os
    import pathlib
    
    # Clone the tensorflow models repository if it doesn't already exist
    if "models" in pathlib.Path.cwd().parts:
      while "models" in pathlib.Path.cwd().parts:
        os.chdir('..')
    elif not pathlib.Path('models').exists():
      !git clone --depth 1 https://github.com/tensorflow/models

• 第2步：

# Install the Object Detection API
%%bash
cd models/research/
protoc object_detection/protos/*.proto --python_out=.
cp object_detection/packages/tf2/setup.py .
python -m pip install .