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

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

ナンバープレート検出と認識AIのプロトタイプ🚙

タイトルにプロトタイプとありますが、永遠に正式版は作られません(たぶん

YOLOの新し目のやつを試してみたかったのと、ChatGPTにモデル構造作らせてみたかったのと、でテキトーに作ったリポジトリです🦔

検出用のデータセットは高橋宅周辺、認識用のデータセットGoogle画像検索です。
そのため、撮影条件などによって著しく検出精度・認識精度が落ちる可能性があるため、もし使用する場合には注意ください。

また、ナンバープレート認識と言うちょっと取り扱いが悩ましいタスクのため、ライセンスはAGPLとしています。

 

2ステージ構造で、ナンバープレート検出(PlateYOLO-JP)、ナンバープレート認識(EkMixer)の2つのモデルで動作しています。
ちなみに、EkMixer と言うモデルは、モデル構造もモデル名もChatGPTがテキトーに考えたやつです👀
正直言えば、その昔僕が作ったモデルのほうが性能良かったのですが、今回はChatGPTに任せてみたかったので、EkMixerを採用してトレーニングしてみました🦔

 

ソースコードや重みは以下のリポジトリで公開しています👻

github.com