1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorch による効率的な AI モデルトレーニング
Connected

演習

画像データセットの前処理

農業向けの精密モニタリングシステムを開発しています。事前学習済みのトランスフォーマーモデルを使って作物の健康状態を監視し、後で農業画像に合わせてファインチューニングする予定です。まず AutoImageProcessor を使ってデータセットを前処理し、トレーニングの準備を整えましょう。

以下のデータはあらかじめ読み込まれています。

  • AutoImageProcessor クラスは transformers からインポート済みです
  • model は microsoft/swin-tiny-patch4-window7-224 に設定されています
  • サンプルの dataset が定義されており、サンプル画像が変数 image に読み込まれています

指示

100 XP
  • 事前定義された model から事前学習済みの画像プロセッサを読み込みます。
  • image_processor をデータセット全体にマッピングします。