Bounding boxes prediction
आपने एक object recognition मॉडल प्रशिक्षित किया है. अब, आप इसे उपयोग करके किसी टेस्ट इमेज के लिए bounding boxes और classifications जनरेट करना चाहते हैं.
मॉडल model के रूप में उपलब्ध है और यह पहले से evaluation मोड में है. test_image भी उपलब्ध है, और torch तथा torch.nn nn नाम से इम्पोर्ट किए गए हैं.
यह अभ्यास पाठ्यक्रम का हिस्सा है
PyTorch के साथ इमेज के लिए डीप लर्निंग
अभ्यास निर्देश
test_imageके लिए मॉडल का आउटपुट लें और उसेoutputमें असाइन करें.- आउटपुट से प्रेडिक्टेड bounding boxes निकालें और उन्हें
boxesमें असाइन करें. - आउटपुट से प्रेडिक्टेड scores निकालें और उन्हें
scoresमें असाइन करें.
इंटरैक्टिव व्यावहारिक अभ्यास
इस अभ्यास को इस नमूना कोड को पूरा करके आज़माएँ।
# Get model's prediction
with torch.no_grad():
output = ____
# Extract boxes from the output
boxes = ____
# Extract scores from the output
scores = ____
print(boxes, scores)