事前学習済みモデルを読み込む
ソーシャルメディアの画像にラベルを付けるアプリケーションを作成しています。この作業には高い精度とスピードが求められます。事前学習済みの ResNet18 モデルを使って、画像クラスを推論します。
この演習はコースの一部です
PyTorch で学ぶ画像向け Deep Learning
演習の手順
torchvision.modelsからresnet18とResNet18_Weightsをインポートします。weightsを重みパラメータとして設定し、resnet18()を使ってmodelをインスタンス化します。modelを評価モードに設定します。- 入力用の変換を初期化し、
transformに代入します。
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
# Import resnet18 model
from torchvision.models import ____
# Initialize model with default weights
weights = ResNet18_Weights.DEFAULT
model = ____
# Set model to evaluation mode
model.____
# Initialize the transforms
transform = ____