कन्वोल्यूशनल नेटवर्क बनाना
आप एक वेदर फोरकास्टिंग सिस्टम बनाने वाली टीम में हैं। सिस्टम के हिस्से के रूप में, अलग-अलग जगहों पर कैमरे लगाए जाएँगे जो आसमान की तस्वीरें लेंगे। आपका काम इन तस्वीरों में अलग-अलग बादल प्रकारों को क्लासिफाई करने वाला मॉडल बनाना है, जो आते हुए वेदर फ्रंट्स को पहचानने में मदद करेगा।
आप एक कन्वोल्यूशनल इमेज क्लासिफायर बनाने का निर्णय लेते हैं। मॉडल दो हिस्सों में होगा:
- एक फीचर एक्सट्रैक्टर, जो इनपुट इमेज से फीचर्स का एक वेक्टर सीखता है,
- एक क्लासिफायर, जो सीखे गए फीचर्स के आधार पर इमेज की क्लास की भविष्यवाणी करता है।
दोनों torch और torch.nn as nn आपके लिए पहले से इम्पोर्ट किए जा चुके हैं, तो आइए शुरू करते हैं!
यह अभ्यास पाठ्यक्रम का हिस्सा है
इंटरमीडिएट Deep Learning with PyTorch
इंटरैक्टिव व्यावहारिक अभ्यास
इस अभ्यास को इस नमूना कोड को पूरा करके आज़माएँ।
class Net(nn.Module):
def __init__(self, num_classes):
super().__init__()
# Define feature extractor
self.feature_extractor = nn.Sequential(
nn.Conv2d(3, 32, kernel_size=3, padding=1),
nn.ELU(),
nn.MaxPool2d(kernel_size=2),
____,
____,
____,
____,
)