先週までにやったこと:
ViT を使った画像分類モデルの動作確認
今週やったこと:
分類モデルで画像を扱えるようにデータの仕分けを行うプログラムを作成しています。糖度データがCSVファイルになっているので、画像データと糖度データを対応させて一定の糖度によってクラス分けを行なっています。
既存モデルでの実験については、先輩のプログラムを動かしたところヒストグラムの読み込みでクラッシュしてしまいその後の動作が確認できなくなってしまったので原因を探りたいと思います。
また、先輩のモデルでは評価データと実験データの仕分けはランダムで行なっているため、すべてのプログラムを動作させるのではなくて、昨年度までの既存モデルに追加学習させる形というのも検討していきます。
来週やること:
ViT分類モデルでの学習
中間発表の資料作成
コメントを残す