बॉक्स रेग्रेसर ब्लॉक
आपका अंतिम कार्य बाउंडिंग बॉक्स के निर्देशांक प्रेडिक्ट करने के लिए एक रेग्रेसर ब्लॉक बनाना है। आपने तय किया है कि इसमें 2 फुली-कनेक्टेड लेयर्स होंगी जिनके बीच में ReLU एक्टिवेशन होगा, ठीक वैसे ही जैसे आपने पहले क्लासिफायर में किया था।
आपका vgg_model और input_dim अब भी उपलब्ध है और torch तथा torchvision.models इम्पोर्ट किए जा चुके हैं।
यह अभ्यास पाठ्यक्रम का हिस्सा है
PyTorch के साथ इमेज के लिए डीप लर्निंग
अभ्यास निर्देश
- एक वैरिएबल
num_coordinatesबनाएँ जिसमें प्रेडिक्ट किए जाने वाले बाउंडिंग बॉक्स निर्देशांकों की संख्या हो। - पहली लीनियर लेयर के लिए उपयुक्त इनपुट डाइमेंशन परिभाषित करें और आउटपुट डाइमेंशन
32सेट करें। - रेग्रेसर की अंतिम लेयर में उपयुक्त आउटपुट डाइमेंशन परिभाषित करें।
इंटरैक्टिव व्यावहारिक अभ्यास
इस अभ्यास को इस नमूना कोड को पूरा करके आज़माएँ।
# Define the number of coordinates
____
bb = nn.Sequential(
# Add input and output dimensions
nn.Linear(____, ____),
nn.ReLU(),
# Add the output for the last regression layer
nn.Linear(32, ____),
)