【機械学習】Google Colabで物体検出を簡単に試す方法

こんばんはエンジニアの眠れない夜です。

今回はGoogle Colabで１行もコードを書かずに物体検出ができるColabを作成しました。

こちらのからColabを開いて「すべてのセルを実行」するだけです。

こちらの記事を参考にさせていただきました。そのままでは情報が古くなっているのか動かなかったので調整しています。

https://qiita.com/nanako_ut/items/cb28d403fcaf013b2396

物体検出のライブラリはMask_RCNNを使っています。

Mask-R-CNNの実行結果

こちらを実行すると物体検出の結果が返ってきます。

Processing 1 images
image                    shape: (425, 640, 3)         min:    0.00000  max:  255.00000  uint8
molded_images            shape: (1, 1024, 1024, 3)    min: -123.70000  max:  150.10000  float64
image_metas              shape: (1, 93)               min:    0.00000  max: 1024.00000  float64
anchors                  shape: (1, 261888, 4)        min:   -0.35390  max:    1.29134  float32

その情報を画像に書き込むとこんな感じで色を付けてくれます。