今週やったこと
Google Colab上でYOLOv5の転移学習をしました。
- モデル：YOLOv5s(一番小さい)
- 学習用画像：210枚
- バッチサイズ：16
- エポック(学習回数)：100
最終的なmAPは0.77でしたが、実際に検出を試してみると、全体が写っている硬貨でも検出漏れがかなり目立っていました。
光の当たり方によって硬貨の色合いが変わっている場合は精度が良くなく、特に100円玉は再現率が0.65と他に比べてかなり低い数値でした。
今回は検出するクラス数に対して学習画像が少なかったことや、ウェブスクレイピングで収集した画像ということもあって学習に適した画像が少なかったこともあったので、これを踏まえて自分で適した画像を撮影するように改善します。
来週やること
- 「形状変形を許容するカメラ画像からの国旗検索に関する研究 | 西部, 福谷」(平成30年)の論文を読む
- 学習用画像の収集