1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorch で学ぶ画像向け Deep Learning
Connected

演習

事前学習済みモデルを読み込む

ソーシャルメディアの画像にラベルを付けるアプリケーションを作成しています。この作業には高い精度とスピードが求められます。事前学習済みの ResNet18 モデルを使って、画像クラスを推論します。

指示

100 XP
  • torchvision.models から resnet18 と ResNet18_Weights をインポートします。
  • weights を重みパラメータとして設定し、resnet18() を使って model をインスタンス化します。
  • model を評価モードに設定します。
  • 入力用の変換を初期化し、transform に代入します。