Încărcarea unui model pre-antrenat

Constructiești o aplicație pentru etichetarea imaginilor din rețelele de socializare. Această sarcină necesită precizie ridicată și viteză de procesare. Vei folosi un model pre-antrenat ResNet18 pentru a clasifica imaginile.

Acest exercițiu face parte din cursul

Deep Learning pentru imagini cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

Importă resnet18 și ResNet18_Weights din torchvision.models.
Instanțiază model folosind resnet18(), setând parametrul weights la weights.
Setează model în modul de evaluare.
Inițializează transformările de intrare și atribuie-le variabilei transform.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

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

Editează și rulează codul