YOLOv3を用いて学習させた検出器で、馬の顔領域を検出したいと考えています。一つの画像から1頭のみを検出したく、検出されたバウンディングボックスのうち、最も大きなものだけを表示させる方法を知りたいのですがどなたかご教授頂けないでしょうか。

Question

隆人山田 2022-1-7

评论：隆人山田 2022-1-7

この画像から、手前の2頭が検出されてしまいます。手前の茶色馬だけを検出する方法を教えていただきたいです。

Answer 1

Shunichi Kusano 2022-1-7

バウンディングボックスには幅と高さの情報が入っています。また、バウンディングボックスと一緒にscoreという検出の信頼性みたいな値も出てきます。

[bbox,score,label] = detect(yolonet,img);

最も大きいという基準で選択したいなら

area = bbox(:,3) .* bbox(:,4);
[~,Idx] = max(area);
bbox1 = bbox(Idx,:);

最も信頼性が高い、なら

[~,Idx] = max(score);
bbox1 = bbox(Idx,:);

といった感じで画像から一つだけ取り出すことができます。

Shunichi Kusano 2022-1-7

エラーメッセージの一番上を見ると原因がわかります。

LabelとPositionの数が合ってない、ということで、bbox1は複数の中から1つのボックスを選んだものなのに対して、scoreは元のまま（複数）になっているのが原因です。

scoreからもIdxで1つ値を取り出して差し替えてあげれば解決すると思います。

隆人山田 2022-1-7

出来ました！！

本当にありがとうございました

また機会があればよろしくお願いします！！