Memuat model pra-latih

Anda sedang membangun sebuah aplikasi untuk memberi label pada gambar dari media sosial. Tugas ini memerlukan akurasi dan kecepatan tinggi. Anda akan menggunakan model ResNet18 yang sudah pra-latih untuk melakukan inferensi kelas gambar.

Latihan ini adalah bagian dari kursus

Deep Learning untuk Gambar dengan PyTorch

Lihat Kursus

Petunjuk latihan

Impor resnet18 dan ResNet18_Weights dari torchvision.models.
Instansiasi model menggunakan resnet18(), dengan parameter weights diatur ke weights.
Atur model ke mode evaluasi.
Inisialisasi transformasi masukan dan tetapkan ke transform.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# 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 = ____

Edit dan Jalankan Kode