今週やったこと
照明条件を変更して硬貨の撮影を行いました。
- 背景色 : 黒(布)
- 照明 : 通常,寒色,暖色
- (5円玉, 10円玉など、酸化によって色の違いが大きいものは2倍の撮影数とする)
- 合計192枚
また、これまでに使用していた(上で撮影した画像は含まない)データセットでYOLOv7の再学習を行いました。
- モデル : yolov7x.pt(2番目に軽量なモデル)
- 学習回数 : 200回
- バッチサイズ : 16
5円玉と10円玉以外の正解率がかなり高いので、今後はYOLOv7を採用しようと考えています。
今後はAlbumentationなどでのデータ拡張も考えていますが、YOLOv7で実装できない場合は、引き続きYOLOv5を使用します。
来週やること
- 学習画像の追加
- YOLOv7で再学習
- 紙幣と硬貨の平行検出を試す