五島x AI: 魚をyolo5で検出してみた。

YOLO v5を用いてカスタムデータセットを用いて物体検出する記事を拝見して、五島でとった魚を物体検出してみた(推論のみ)。

Training a Custom Object Detection Model With Yolo-V5 | by Prabhat Kumar Sahu | Analytics Vidhya | Jan, 2021 | Medium

YOLOv5とは

入力画像から物体の位置と種類を検出することができます。

実験コード

下記に保存しました。

github.com

結果

検出できなかった魚

やっぱり、学習していない形の魚だとだめですねー。

こんな感じにしたいなぁ。アノテーション頑張ってみようかなぁ。。。 https://i.imgur.com/s4PgS4X.gif

今度、帰省したときは、写真撮影に励もうと思う。