AIで製品の個数を数える方法

Question

実現したいこと

下図のように製品の個数をAIで数えるプログラムを作りたいです。

count = 5

発生している問題・分からないこと

下記①、②の操作はできるようになりましたが、
学習させたモデルを使って複数製品を数える方法がわかりません。

①CNNでねじ画像を学習させ、ねじかねじ以外かを認識させる

↓学習させた画像の例

②Opencvを用いて製品の数を数える
グレースケール→2値化→輪郭を描く→特定の面積以上の個数をカウント

最初は②の手法のみで進めていましたが、検出精度が上がらないことと、
将来的に不良品の検出も行いたいため、①の手法も進めています。

AIで製品の個数を数えるには、
①と②の組み合わせで実現できると思っていますが、
具体的にどのように実装したらいいのかわかりません。

該当のソースコード

特になし

試したこと・調べたこと

• teratailやGoogle等で検索した
• ソースコードを自分なりに変更した
• 知人に聞いた
• その他

上記の詳細・結果

Yoloを用いて認識した物体にバウンディングボックスを描く。
という操作ができる事はわかりました。

しかし、私が実現したいのは、生産ラインで特定の製品の個数を数えるという操作であるため
個別に製品画像を学習させる必要があります。

補足

Python 3.8.3

Answer 1

OpenCVを使うことはなく、物体検出 / 物体検知 AIでやる領域でしょうね。
これらのワードで検索すると沢山の日本語記事がヒットしますので、参考にするといいでしょう。

ざっと見て以下のサイトが面白そうでした。
https://farml1.com/yolov5/

Comments

[python01]

ご回答ありがとうございます！

面白そうな題材なので、理解し易そうですね。確認してみます！

[python01]

一番下に動画がありますが、まさにこの動画のような動作を実現したいと思っています。
試してみます

Answer 2

複数の物体検出ですね。
Faster R-CNNやYOLOとかの論文を読むことをお勧めします。
https://arxiv.org/pdf/1506.01497.pdf
https://arxiv.org/pdf/1506.02640.pdf

Comments

[python01]

ご回答ありがとうございます！
論文を確認してみます。