google colabでtensorflowのobject_detectionモジュールを使えるようになるまで

覚書なので、全くもって完ぺきではありません。なのでより良い方法がありましたら、コメントをお願いします。

参考　https://github.com/tensorflow/models/blob/master/research/object_detection/colab_tutorials/inference_tf2_colab.ipynb

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

ここでhttps://github.com/tensorflow/modelsをクローンする。

# 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 .

object_detectionモジュールのセットアップ。10分ぐらいかかる。

from object_detection.utils import label_map_util

これでobject_detectionモジュールを使用できる。