始める無料で始める

画像分類

この演習では、前と同じ flickr データセット(3万枚の画像とそのキャプション)を使います。今回は、事前学習済みモデルを使って画像を分類します。

Photo of 2 people, 1 is playing the guitar

サンプル画像(image)とパイプラインモジュール(pipeline)は読み込まれています。

この演習はコースの一部です

Hugging Face で学ぶマルチモーダルモデル

コースを見る

演習の手順

  • 事前学習済みモデル google/mobilenet_v2_1.0_224 を使って、image-classification パイプラインを読み込みます。
  • パイプラインを使ってサンプル画像のクラスを予測します。
  • 最初の(最も確率が高い)label を出力します。

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

# Load the image-classification pipeline
pipe = ____

# Use the pipeline to predict the class of the sample image
pred = ____

# Print the first (highest probability) label
print("Predicted class:", ____)
コードを編集して実行