提供されたアノテーションデータの学習

[ShiraiSoma]

Honda提供のアノテーションデータについて，学習と推論を試している．
学習時のパラメータとしてはデータ拡張など基本的にYOLOPのものから変えていない．
バッチサイズ，エポック数，画像サイズを変更している．
現状として，以下のような結果が得られている．

• エポック数:240, バッチサイズ:16,画像サイズ[640,640]

• エポック数:500, バッチサイズ:24，画像サイズ[1280,840]

現在進めている事

• データ拡張のパラメータを変更
• アノテーションデータのマスク画像を調整

[kyo0221]

バッチサイズ16はかなり少ない気がしますね。
過学習のリスクがあると思います。

訓練データとテストデータで分けて用意して、過学習しているか確認しても良いかもしれません。

[kyo0221]

バッチサイズは一般的に2のn乗で定義されることが多いので、少ないバッチ数で検証するのであれば32や64などで学習すると良いと思います。

[ShiraiSoma]

頂いたコメントから，バッチサイズを64にして試している．

• エポック数:500, バッチサイズ:64，画像サイズ[640,640]

また，データ拡張のパラメータを変更して試してもいる．

• 変更点
  • スケーリング範囲:0.25 → 0.05, 回転範囲:10 → 2, RGB変換:False → True
• エポック数:500, バッチサイズ:16，画像サイズ[640,640]