Tensorflowを用いたオブジェクト検出

以前、オブジェクト検出AIのトレーニングを行いました。

目次

ある寒い冬の7月の日… その日はオーストラリアにいた… 私は、未封のコンクリート補強筋を検出するためのAIモデルを訓練するという緊急の必要性を感じた…

そのため、地元のBunnings Warehouse店からいくつかの新しい輝く補強筋バーを購入した

concrete reinforcement bars

そして、このような補強筋バー用の安全キャップも購入した

concrete reinforcement bar caps

キャップから始めた

これらの黄色の補強筋キャップを使って、さまざまな構成の写真を100枚撮影し、LabelImgでラベル付けした。その後、SSDモデルを使用し、またEfficientNetを使ってPythonで少しコードを書いた。すると、モデルは私が望んでいた通りに素早く学習を始めた。

reo bar cap - lh

reo bar cap - rh

当初、モデルはほぼすべての黄色い円筒形の物体を補強筋キャップとして認識した

coffee cap

そして、バウンディングボックスはもっと正確にできるはずだった…

次にバーが登場

ガレージからいくつかの他の便利な工具を取り出し、このような写真をもう100枚撮影した。すると、モデルがよりよく認識し始めた…

garage tools object detection

はい、明らかに改善されている。

garage tools object detection

ai object detection training in the garage

モバイルアプリ

最終的に、私のモバイル電話で巨大な遅延なしに動作する小さなSSD AIモデルを訓練し、シンプルなAndroidアプリを作成した。それがSamsung S8でどのように見えるかはこちらです:

mobile phone ssd ai

mobile phone object detection

mobile tensorflow

これは非常に良い実験だった。

有用なリンク