データ拡張つきのデータセット

雲の画像からデータセットを作成し、さまざまな雲の種類を分類するための畳み込みモデルはすでに用意されています。学習を始める前に、モデルの雲分類の精度向上が見込めるデータ拡張をデータセットに加えていきましょう。

Dataset と DataLoader のセットアップ用コードはすでに用意されており、見覚えがあるはずです。あなたのタスクは、画像の読み込み時に適用する変換の合成を定義することです。

これまで表示を見やすくするために画像を 128×128 にリサイズしていましたが、今回は学習を高速化するためにより小さいサイズを使います。後ほど確認しますが、64×64 でもモデルが学習するには十分です。