Google MobileObjectLocalizerを用いた物体位置検出

Google MobileObjectLocalizerと言うモデルがTensorFlow-Hubで公開されていたため、試してみました🦔

上記ページによると、以下のような特徴を持つモデルのようです。

MobileObjectLocalizerはクラスに依存しないオブジェクト検出器です。
オブジェクト分類は無く、検出されるクラスは1種類(Entity)のみです。
このモデルは、画像内の最も目立つ物体を検出するのに適しています。
自動運転のための障害物や人間の検出などのミッションクリティカルな使用には適していません。

以下のような感じで動作します。

箸置きはCOCOで訓練された物体検出モデルでは検出されない物体です👻

改めてMobile Object LocalizerをONNXに変換してお試し👀
結構それっぽいぞー👻 pic.twitter.com/2QJaITE4hN

— 高橋 かずひと@孫請級プログラマー🦔 (@KzhtTkhs) 2021年9月19日

今回試したソースコードは整理して以下にコミットしています。