1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorchによる中級ディープラーニング
Connected

演習

PyTorch によるデータ拡張

Dataset にデータ拡張を組み込み、変換が期待どおりに適用されているかを画像で確認してみましょう。

まず、train_transforms に拡張用の変換を追加します。ここでは、ランダムな水平方向のフリップと、0〜45度の範囲でランダムに回転させます。続く Dataset と DataLoader の作成コードは、これまでとまったく同じです。最後に、画像をリシェイプして表示し、新しい拡張の効果が見えるか確認します。

必要なインポートはあらかじめ用意されています。

from torchvision.datasets import ImageFolder
from torch.utils.data import DataLoader
from torchvision import transforms
import matplotlib.pyplot as plt

さあ、雲の写真でデータ拡張を試してみましょう！

指示

100 XP
  • train_transforms に、ランダムな水平方向のフリップと、0〜45度の範囲でランダムに回転させる変換をこの順で追加します。
  • DataLoader から取り出した image テンソルを、表示に適した形にリシェイプします。
  • 画像を表示します。