Pre-trained मॉडल लोड करना
आप सोशल मीडिया से आने वाली images को लेबल करने के लिए एक एप्लिकेशन बना रहे हैं. इस काम में high accuracy और speed चाहिए. आप image classes infer करने के लिए pre-trained ResNet18 मॉडल का उपयोग करने वाले हैं.
यह अभ्यास पाठ्यक्रम का हिस्सा है
PyTorch के साथ इमेज के लिए डीप लर्निंग
अभ्यास निर्देश
torchvision.modelsसेresnet18औरResNet18_Weightsइम्पोर्ट करें.resnet18()का उपयोग करकेmodelको instantiate करें और weights पैरामीटर कोweightsपर सेट करें.modelको evaluation मोड में सेट करें.- input transforms इनिशियलाइज़ करें और उन्हें
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 = ____