高橋かずひとのプログラミング、その他、備忘録。

日々調べてたことや、作ってみたものをメモしているブログ。　お決まりの断り文句ですが、このブログに書かれている内容は個人の見解であり、所属する組織の公式見解ではありません。チラ裏。

OpenLendaで信号機検出をお試し🚥

機械学習 物体検出

TuringさんがOpenLendaと言う信号機検出のモデルを公開されていたので、試してみました👀

HugginFaceでデモが公開されているので、さっと試すことができます。

huggingface.co

 

モデルやPyTorchの推論コードは公式リポジトリで公開されています。

pthモデルの他にonnxファイルも公開されていて、地味にうれしいです🦔

github.com

 

ONNXが公開されているのであれば、それで推論したい。と言うことでザクッとサンプルプログラムを作成しました。

マルチクラスNMSのちょうど良い参考資料が無くて、すっごい暫定実装です👻

github.com

 

動かした感じは以下↓