1. 学ぶ
  2. /
  3. コース
  4. /
  5. Hugging Face で学ぶマルチモーダルモデル

Connected

演習

CVのファインチューニング:モデルクラス

この演習では、事前学習済みモデルを読み込み、元の ImageNet の 1000 クラスではなく、Stanford Cars データセット に含まれる新しい「車種」分類に合わせて出力を調整します。データセットには車のラベル付き画像が含まれています。

データセット(dataset)と、transformers の AutoModelForImageClassification はすでに読み込まれています。データセットは、3 つの車種のみを含むようにフィルタ済みです。

指示

100 XP
  • データセットから新しいラベル名を取得します。
  • モデルの読み込み時に、新しい id2label マッピングを追加します。
  • 対応する label2id マッピングを追加します。
  • クラス数を変更するために必要なフラグを追加します。